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