16 lines
393 B
Plaintext
16 lines
393 B
Plaintext
name: s487197
|
|
|
|
docker_env:
|
|
image: s487197/ium:56
|
|
|
|
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}" |