ium_487187/MLproject

13 lines
247 B
Plaintext
Raw Normal View History

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}"