1  name: pytorch_tutorial
 2  
 3  python_env: python_env.yaml
 4  
 5  entry_points:
 6    main:
 7      parameters:
 8        batch-size: {type: int, default: 64}
 9        test-batch-size: {type: int, default: 1000}
10        epochs: {type: int, default: 10}
11        lr: {type: float, default: 0.01}
12        momentum: {type: float, default: 0.5}
13        enable-cuda: {type: string, default: 'True'}
14        seed: {type: int, default: 5}
15        log-interval: {type: int, default: 100}
16      command: |
17            python mnist_tensorboard_artifact.py \
18              --batch-size {batch-size} \
19              --test-batch-size {test-batch-size} \
20              --epochs {epochs} \
21              --lr {lr} \
22              --momentum {momentum} \
23              --enable-cuda {enable-cuda} \
24              --seed {seed} \
25              --log-interval {log-interval}