11 lines
296 B
Plaintext
11 lines
296 B
Plaintext
|
name: Fifa Players
|
||
|
docker_env:
|
||
|
image: docker.io/adnovac/ium_s434760:2.0
|
||
|
entry_points:
|
||
|
train:
|
||
|
parameters:
|
||
|
batch_size: {type: int, default: 15}
|
||
|
epochs: {type: int, default: 16}
|
||
|
command: "python train.py {batch_size} {epochs}"
|
||
|
evaluate:
|
||
|
command: "python evaluate.py"
|