| python train_unet.py \ | |
| --dataset imagenette \ | |
| --data-root ./data \ | |
| --imagenette-split train \ | |
| --single-class \ | |
| --class-id 1 \ | |
| --epochs 100 \ | |
| --batch-size 16 \ | |
| --config unet_cifar.yaml \ | |
| --save-dir runs/cfm_unet/imagenette1_128_100ep \ | |
| --log-dir runs/cfm_unet/imagenette1_128_100ep/tb \ | |