name: s487176
docker_env:
  image: mactyc/ium
entry_points:
  train:
    parameters: 
      num_epochs {type: int, default: 10}
      lr : {type: int, default : 0.01}
      alpha: {tpe : int, default: 0.01}
    command: "python zad1.py --num_epochs {num_epochs} --lr {lr}"