Upload folder using huggingface_hub
Browse files- 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 |
-
|
| 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):
|