ium_487187/MLproject
Jakub Zaręba 04fe4bb719 ss
2023-05-10 23:54:20 +02:00

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