update config

mindspore-lab · Sep 27, 2024 · 6d06bfa · 6d06bfa
1 parent 65baffc
commit 6d06bfa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/opensora_hpcai/configs/vae/train/stage1.yaml b/examples/opensora_hpcai/configs/vae/train/stage1.yaml
@@ -26,7 +26,7 @@ micro_batch_size: null
 # training recipe
 seed: 42
 use_discriminator: False
-dtype: "fp16"
+dtype: "bf16"
 batch_size: 1
 clip_grad: True
 max_grad_norm: 1.0

diff --git a/examples/opensora_hpcai/configs/vae/train/stage3.yaml b/examples/opensora_hpcai/configs/vae/train/stage3.yaml
@@ -17,7 +17,7 @@ dataset_name: "video"
 csv_path: "../videocomposer/datasets/webvid5_copy.csv"
 video_folder: "../videocomposer/datasets/webvid5"
 frame_stride: 1
-num_frames: 33  # TODO: set 33 after dynamic shape adaptation and posterior concat fixed
+num_frames: 17  # TODO: set 33 after dynamic shape adaptation and posterior concat fixed
 image_size: 256
 
 micro_frame_size: 17
@@ -27,7 +27,7 @@ micro_batch_size: 4
 # training recipe
 seed: 42
 use_discriminator: False
-dtype: "fp16"
+dtype: "bf16"
 batch_size: 1
 clip_grad: True
 max_grad_norm: 1.0