Xsmos commited on
Commit
d08708f
·
verified ·
1 Parent(s): 0a23c67
Files changed (2) hide show
  1. diffusion.py +1 -1
  2. quantify_results.ipynb +0 -0
diffusion.py CHANGED
@@ -586,7 +586,7 @@ def train(rank, world_size):
586
 
587
  ddp_setup(rank, world_size)
588
 
589
- num_train_image_list = [3200]#[200]#[1600,3200,6400,12800,25600]
590
  for i, num_image in enumerate(num_train_image_list):
591
  config.num_image = num_image
592
  # config.world_size = world_size
 
586
 
587
  ddp_setup(rank, world_size)
588
 
589
+ num_train_image_list = [200]#[3200]#[200]#[1600,3200,6400,12800,25600]
590
  for i, num_image in enumerate(num_train_image_list):
591
  config.num_image = num_image
592
  # config.world_size = world_size
quantify_results.ipynb CHANGED
The diff for this file is too large to render. See raw diff