Yash Nagraj commited on
Commit ·
0313c13
1
Parent(s): e9c60a4
Put it to eval mode and sampeled some images
Browse files
SampledImgs/SampledNoGuidenceImgs1.png
ADDED
|
__pycache__/diffusion.cpython-312.pyc
CHANGED
|
Binary files a/__pycache__/diffusion.cpython-312.pyc and b/__pycache__/diffusion.cpython-312.pyc differ
|
|
|
__pycache__/model.cpython-312.pyc
CHANGED
|
Binary files a/__pycache__/model.cpython-312.pyc and b/__pycache__/model.cpython-312.pyc differ
|
|
|
main.py
CHANGED
|
@@ -3,7 +3,7 @@ from train import train, eval
|
|
| 3 |
|
| 4 |
def main(model_config=None):
|
| 5 |
modelConfig = {
|
| 6 |
-
"state": "
|
| 7 |
"epochs": 200,
|
| 8 |
"batch_size": 80,
|
| 9 |
"T": 1000,
|
|
@@ -20,11 +20,11 @@ def main(model_config=None):
|
|
| 20 |
"grad_clip": 1.,
|
| 21 |
"device": "cuda:0",
|
| 22 |
"training_load_weight": None,
|
| 23 |
-
"checkpoint_dir": "
|
| 24 |
-
"test_load_weight": "
|
| 25 |
-
"
|
| 26 |
"sampledNoisyImgName": "NoisyNoGuidenceImgs.png",
|
| 27 |
-
"sampledImgName": "
|
| 28 |
"nrow": 8
|
| 29 |
}
|
| 30 |
|
|
|
|
| 3 |
|
| 4 |
def main(model_config=None):
|
| 5 |
modelConfig = {
|
| 6 |
+
"state": "eval", # or eval
|
| 7 |
"epochs": 200,
|
| 8 |
"batch_size": 80,
|
| 9 |
"T": 1000,
|
|
|
|
| 20 |
"grad_clip": 1.,
|
| 21 |
"device": "cuda:0",
|
| 22 |
"training_load_weight": None,
|
| 23 |
+
"checkpoint_dir": "Checkpoints/",
|
| 24 |
+
"test_load_weight": "ckpt_199.pth",
|
| 25 |
+
"sample_dir": "SampledImgs/",
|
| 26 |
"sampledNoisyImgName": "NoisyNoGuidenceImgs.png",
|
| 27 |
+
"sampledImgName": "SampledNoGuidenceImgs1.png",
|
| 28 |
"nrow": 8
|
| 29 |
}
|
| 30 |
|