36c95ba
1
2
3
4
5
6
7
8
data_path: ./data batch_size: 4 num_epochs: 2 lr: 1e-3 output_path: ./output image_size: [224, 224]
