Upload folder using huggingface_hub
Browse files- checkpoint/epoch_2nd_00000.pth +3 -0
- config.yml +4 -4
- epoch_1st_00000.pth +2 -2
- tensorboard/events.out.tfevents.1742831193.a31e41537750.4050.0 +3 -0
- train.log +0 -0
checkpoint/epoch_2nd_00000.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:417add804578d77a85a35674855ac0d40e197a55dbe1a809d83ec367b8a62bb3
|
| 3 |
+
size 1499858424
|
config.yml
CHANGED
|
@@ -7,11 +7,11 @@ epochs_1st: 20 # number of epochs for first stage training (pre-training)
|
|
| 7 |
epochs_2nd: 100 # number of peochs for second stage training (joint training)
|
| 8 |
# first epoch = 3x650
|
| 9 |
# second epoch = 2x440
|
| 10 |
-
|
| 11 |
batch_size: 2
|
| 12 |
-
max_len:
|
| 13 |
pretrained_model: "epoch_1st_00000.pth"
|
| 14 |
-
second_stage_load_pretrained:
|
| 15 |
load_only_params: false # set to true if do not want to load epoch numbers and optimizer parameters
|
| 16 |
|
| 17 |
F0_path: "Utils/JDC/bst.t7"
|
|
@@ -101,7 +101,7 @@ loss_params:
|
|
| 101 |
lambda_diff: 1. # score matching loss (2nd stage)
|
| 102 |
|
| 103 |
diff_epoch: 0 # style diffusion starting epoch (2nd stage)
|
| 104 |
-
joint_epoch:
|
| 105 |
|
| 106 |
optimizer_params:
|
| 107 |
lr: 0.0001 # general learning rate
|
|
|
|
| 7 |
epochs_2nd: 100 # number of peochs for second stage training (joint training)
|
| 8 |
# first epoch = 3x650
|
| 9 |
# second epoch = 2x440
|
| 10 |
+
# second stage 1st epoch = 2x300
|
| 11 |
batch_size: 2
|
| 12 |
+
max_len: 305 # maximum number of frames
|
| 13 |
pretrained_model: "epoch_1st_00000.pth"
|
| 14 |
+
second_stage_load_pretrained: true # set to true if the pre-trained model is for 2nd stage
|
| 15 |
load_only_params: false # set to true if do not want to load epoch numbers and optimizer parameters
|
| 16 |
|
| 17 |
F0_path: "Utils/JDC/bst.t7"
|
|
|
|
| 101 |
lambda_diff: 1. # score matching loss (2nd stage)
|
| 102 |
|
| 103 |
diff_epoch: 0 # style diffusion starting epoch (2nd stage)
|
| 104 |
+
joint_epoch: 30 # joint training starting epoch (2nd stage)
|
| 105 |
|
| 106 |
optimizer_params:
|
| 107 |
lr: 0.0001 # general learning rate
|
epoch_1st_00000.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:417add804578d77a85a35674855ac0d40e197a55dbe1a809d83ec367b8a62bb3
|
| 3 |
+
size 1499858424
|
tensorboard/events.out.tfevents.1742831193.a31e41537750.4050.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fc47ed09b85962f915e02da19cebefda7ac84ab0e6b90f59eaaaa4f02a948fbc
|
| 3 |
+
size 156744
|
train.log
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|