name: s464980

conda_env: conda.yaml

entry_points:
  optimal_parameters:
    parameters:
      epochs: { type: int, default: 20 }
    command: 'python train.py {epochs}'