2023-05-10 22:55:38 +02:00
|
|
|
name: s487187-training
|
|
|
|
|
2023-05-10 23:54:20 +02:00
|
|
|
docker_env:
|
|
|
|
image: python:3.11
|
|
|
|
volumes: [ "/root/.cache:/root/.cache" ]
|
|
|
|
user_env_vars: [ "SACRED_IGNORE_GIT" ]
|
|
|
|
|
2023-05-10 22:55:38 +02:00
|
|
|
entry_points:
|
|
|
|
main:
|
|
|
|
parameters:
|
|
|
|
epochs: int
|
2023-05-10 23:54:20 +02:00
|
|
|
command: "python train.py --epochs {epochs}"
|