13 lines
247 B
Plaintext
13 lines
247 B
Plaintext
name: s487187-training
|
|
|
|
docker_env:
|
|
image: python:3.11
|
|
volumes: [ "/root/.cache:/root/.cache" ]
|
|
user_env_vars: [ "SACRED_IGNORE_GIT" ]
|
|
|
|
entry_points:
|
|
main:
|
|
parameters:
|
|
epochs: int
|
|
command: "python train.py --epochs {epochs}"
|