Upload 2 files
Browse files- bm-general.pth +2 -2
- bm-general.yaml +3 -3
bm-general.pth
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c2b5c961d59cb7640f9675f362b4cc36277ff5e8017b300dbd18a24bdf40980a
|
| 3 |
+
size 188001914
|
bm-general.yaml
CHANGED
|
@@ -30,7 +30,7 @@ frame_num:
|
|
| 30 |
lmdb: true
|
| 31 |
lmdb_dir: ./datasets/lmdb
|
| 32 |
local_rank: 0
|
| 33 |
-
log_dir: ./logs/training/ucf_2024-09-19-
|
| 34 |
logdir: ./logs
|
| 35 |
loss_func:
|
| 36 |
cls_loss: cross_entropy
|
|
@@ -44,7 +44,7 @@ mean: [0.5, 0.5, 0.5]
|
|
| 44 |
metric_scoring: auc
|
| 45 |
mode: train
|
| 46 |
model_name: ucf
|
| 47 |
-
nEpochs:
|
| 48 |
optimizer:
|
| 49 |
adam:
|
| 50 |
amsgrad: false
|
|
@@ -85,4 +85,4 @@ train_dataset:
|
|
| 85 |
- bitmind/ffhq-256___stable-diffusion-xl-base-1.0
|
| 86 |
with_landmark: false
|
| 87 |
with_mask: false
|
| 88 |
-
workers:
|
|
|
|
| 30 |
lmdb: true
|
| 31 |
lmdb_dir: ./datasets/lmdb
|
| 32 |
local_rank: 0
|
| 33 |
+
log_dir: ./logs/training/ucf_2024-09-19-07-54-06
|
| 34 |
logdir: ./logs
|
| 35 |
loss_func:
|
| 36 |
cls_loss: cross_entropy
|
|
|
|
| 44 |
metric_scoring: auc
|
| 45 |
mode: train
|
| 46 |
model_name: ucf
|
| 47 |
+
nEpochs: 20
|
| 48 |
optimizer:
|
| 49 |
adam:
|
| 50 |
amsgrad: false
|
|
|
|
| 85 |
- bitmind/ffhq-256___stable-diffusion-xl-base-1.0
|
| 86 |
with_landmark: false
|
| 87 |
with_mask: false
|
| 88 |
+
workers: 7
|