name: 434704

docker_env:
    image: jarmosz/ium
entry_points:
    main:
        parameters:
            verbose: {type: int, default: 100}
            epochs: {type: int, default: 0}
        command: "python3 mlflow_exp.py {epochs} {verbose}"