name: s464863

conda_env: conda.yaml

entry_points:
  main:
    parameters:
        learning_rate: {type: float, default: 0.001}
        weight_decay: {type: float, default: 0.001}
        num_epochs: {type: int, default: 1000}
    command: "python ../create_model.py {learning_rate} {weight_decay} {num_epochs}"