Upload folder using huggingface_hub

train_cifar1.sh

@@ -0,0 +1,11 @@
+python train_unet.py \
+        --dataset cifar10 \
+        --data-root ./data \
+        --cifar-split train \
+        --single-class \
+        --class-id 1 \
+        --epochs 100 \
+        --batch-size 2166 \
+        --config unet_cifar.yaml \
+        --save-dir runs/cfm_unet/cifar1_100ep \
+        --log-dir runs/cfm_unet/cifar1_100ep/tb \

train_imagenette1.sh

@@ -0,0 +1,11 @@
+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 \