name: fifa_players

docker_env:
   image: adnovac/ium_s434760

entry_points:
  train:
    parameters:
      batch_size: {type: int, default: 15}
      epochs: {type: int, default: 16}
    command: "python3.8 train.py {batch_size} {epochs}"