bug fix (mindspore-lab#988)

Author: alien-0119

examples/opensora_pku/opensora/models/causalvideovae/model/vae/modeling_wfvae.py

@@ -605,7 +605,7 @@ def sample(self, mean, logvar):
         # sample z from latent distribution
         logvar = mint.clamp(logvar, -30.0, 20.0)
         std = self.exp(0.5 * logvar)
-        z = mean + std * ops.stop_gradient(self.stdnormal(size=mean.shape))
+        z = mean + std * ops.stop_gradient(self.stdnormal(mean.shape))
 
         return z
 

examples/opensora_pku/opensora/train/commons.py

@@ -141,7 +141,7 @@ def parse_train_args(parser):
     parser.add_argument("--dataloader_num_workers", default=12, type=int, help="num workers for dataloder")
     parser.add_argument("--max_rowsize", default=32, type=int, help="max rowsize for data loading")
     parser.add_argument(
-        "dataset_iterator_no_copy",
+        "--dataset_iterator_no_copy",
         default=True,
         type=str2bool,
         help="dataset iterator optimization strategy. Whether dataset iterator creates a Tensor without copy.",

examples/opensora_pku/opensora/train/train_causalvae.py

@@ -191,7 +191,7 @@ def main(args):
         device_num=device_num,
         rank_id=rank_id,
         ds_name="video",
-        dataset_iterator_do_copy=args.dataset_iterator_no_copy,
+        dataset_iterator_no_copy=args.dataset_iterator_no_copy,
     )
     dataset_size = train_loader.get_dataset_size()
 

examples/opensora_pku/opensora/train/train_t2v_diffusers.py

@@ -323,7 +323,7 @@ def main(args):
         collate_fn=collate_fn,
         sampler=sampler,
         column_names=["pixel_values", "attention_mask", "text_embed", "encoder_attention_mask"],
-        dataset_iterator_do_copy=args.dataset_iterator_no_copy,
+        dataset_iterator_no_copy=args.dataset_iterator_no_copy,
     )
     dataloader_size = dataloader.get_dataset_size()
     assert (
@@ -358,7 +358,7 @@ def main(args):
             collate_fn=collate_fn,
             sampler=sampler,
             column_names=["pixel_values", "attention_mask", "text_embed", "encoder_attention_mask"],
-            dataset_iterator_do_copy=args.dataset_iterator_no_copy,
+            dataset_iterator_no_copy=args.dataset_iterator_no_copy,
         )
         val_dataloader_size = val_dataloader.get_dataset_size()
         assert (