dxm21 commited on
Commit
3280f9e
·
verified ·
1 Parent(s): aaedc1e

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. gs/train_vdpm.py +4 -3
gs/train_vdpm.py CHANGED
@@ -564,9 +564,10 @@ class VDPM3DGSTrainer:
564
  pbar.update(1)
565
 
566
  # Save checkpoint
567
- if (it + 1) % 500 == 0 or it == num_iterations - 1:
568
- self.save(it + 1)
569
-
 
570
  print("Training complete!")
571
 
572
  def save(self, iteration: int):
 
564
  pbar.update(1)
565
 
566
  # Save checkpoint
567
+ # if (it + 1) % 500 == 0 or it == num_iterations - 1:
568
+ # self.save(it + 1)
569
+ if it == num_iterations - 1:
570
+ self.save(it + 1)
571
  print("Training complete!")
572
 
573
  def save(self, iteration: int):