fix bug of noise sampling

wangqiang9 · May 11, 2023 · 8250c35 · 8250c35
1 parent 8f8e80d
commit 8250c35
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/sample.py b/sample.py
@@ -7,9 +7,9 @@
     NUM_CLASSES,
     model_and_diffusion_defaults,
     # different modes
-    # create_model_and_diffusion,
+    create_model_and_diffusion,
     # create_model_and_diffusion_acc
-    create_model_and_diffusion_noise,
+    # create_model_and_diffusion_noise,
     add_dict_to_argparser,
     args_to_dict,
 )
@@ -39,10 +39,10 @@ def main():
     logger.configure(args.log_dir)
 
     logger.log("creating model and diffusion...")
-    # different modes
-    #model, diffusion = create_model_and_diffusion(
+    # different modes, if noise or acc method, please specify 'data', 'raster', and 'loss'.
+    model, diffusion = create_model_and_diffusion(
     #model, diffusion = create_model_and_diffusion_acc(
-    model, diffusion = create_model_and_diffusion_noise(
+    #model, diffusion = create_model_and_diffusion_noise(
         **args_to_dict(args, model_and_diffusion_defaults().keys())
     )
     model.load_state_dict(