2021-05-24 00:52:24 +02:00
|
|
|
name: fifa_players
|
|
|
|
|
2021-05-16 23:41:55 +02:00
|
|
|
docker_env:
|
2021-05-24 00:52:24 +02:00
|
|
|
image: adnovac/ium_s434760
|
|
|
|
|
2021-05-16 23:41:55 +02:00
|
|
|
entry_points:
|
|
|
|
train:
|
|
|
|
parameters:
|
|
|
|
batch_size: {type: int, default: 15}
|
|
|
|
epochs: {type: int, default: 16}
|
2021-05-24 00:52:24 +02:00
|
|
|
command: "python3.8 train.py {batch_size} {epochs}"
|