Upload 2 files

bm-general.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b9bb783a13233ed2eccef64d47e10a446e6f90c4f75c6e0e1a8b9d017811ff1a
-size 188008058
+oid sha256:c2b5c961d59cb7640f9675f362b4cc36277ff5e8017b300dbd18a24bdf40980a
+size 188001914

bm-general.yaml

@@ -30,7 +30,7 @@ frame_num:
 lmdb: true
 lmdb_dir: ./datasets/lmdb
 local_rank: 0
-log_dir: ./logs/training/ucf_2024-09-19-03-24-49
+log_dir: ./logs/training/ucf_2024-09-19-07-54-06
 logdir: ./logs
 loss_func:
   cls_loss: cross_entropy
@@ -44,7 +44,7 @@ mean: [0.5, 0.5, 0.5]
 metric_scoring: auc
 mode: train
 model_name: ucf
-nEpochs: 5
+nEpochs: 20
 optimizer:
   adam:
     amsgrad: false
@@ -85,4 +85,4 @@ train_dataset:
 - bitmind/ffhq-256___stable-diffusion-xl-base-1.0
 with_landmark: false
 with_mask: false
-workers: 3
+workers: 7