Update actor_critic_demo.yaml

lisiyao21 · Jun 13, 2022 · 0168365 · 0168365
1 parent 889c9d1
commit 0168365
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/configs/actor_critic_demo.yaml b/configs/actor_critic_demo.yaml
@@ -117,13 +117,13 @@ data:
 testing:
     height: 540
     width: 960
-    ckpt_epoch: 10
+    ckpt_epoch: 5
 
 
 vqvae_weight: 'experiments/sep_vqvae_root/ckpt/epoch_500.pt'
 expname: actor_critic_for_demo_paper
 init_weight: 'experiments/cc_motion_gpt/ckpt/epoch_400.pt'
-epoch: 500
+epoch: 50
 batch_size: 128
 save_per_epochs: 1
 test_freq: 5
@@ -149,4 +149,4 @@ rl_music_root: data/demo_music_feat_60fps
 # sample_shift: 15
 
 # sample_code_length: 150
-# analysis_array: [438, 511, 419, 11, 7, 92, 128, 256, 333]
+# analysis_array: [438, 511, 419, 11, 7, 92, 128, 256, 333]