2023-05-12 15:36:50 +02:00
|
|
|
name: s487197
|
|
|
|
|
|
|
|
|
|
docker_env:
|
2023-05-12 16:03:52 +02:00
|
|
|
image: s487197/ium:56
|
2023-05-12 15:36:50 +02:00
|
|
|
|
|
|
|
|
entry_points:
|
|
|
|
|
main:
|
|
|
|
|
parameters:
|
|
|
|
|
epochs: {type: int, default: 20}
|
|
|
|
|
lr: {type: float, default: 0.01}
|
|
|
|
|
validation_split: {type: float, default: 0.2}
|
|
|
|
|
command: "python ium_train.py {epochs} {lr} {validation_split}"
|
|
|
|
|
test:
|
|
|
|
|
parameters:
|
|
|
|
|
build: {type: int, default: 1}
|
|
|
|
|
command: "python ium_predict.py {build}"
|
