Yash Nagraj commited on
Commit ·
64cb435
1
Parent(s): 00315bf
Sampled different timesteps
Browse files
SampledImgs/SampledNoGuidenceImgs_400.png
ADDED
|
SampledImgs/SampledNoGuidenceImgs_600.png
ADDED
|
main.py
CHANGED
|
@@ -6,7 +6,7 @@ def main(model_config=None):
|
|
| 6 |
"state": "eval", # or eval
|
| 7 |
"epochs": 200,
|
| 8 |
"batch_size": 80,
|
| 9 |
-
"T":
|
| 10 |
"channel": 128,
|
| 11 |
"ch_mult": [1, 2, 3, 4],
|
| 12 |
"attn": [2],
|
|
@@ -24,7 +24,7 @@ def main(model_config=None):
|
|
| 24 |
"test_load_weight": "ckpt_199.pth",
|
| 25 |
"sample_dir": "SampledImgs/",
|
| 26 |
"sampledNoisyImgName": "NoisyNoGuidenceImgs.png",
|
| 27 |
-
"sampledImgName": "
|
| 28 |
"nrow": 8
|
| 29 |
}
|
| 30 |
|
|
|
|
| 6 |
"state": "eval", # or eval
|
| 7 |
"epochs": 200,
|
| 8 |
"batch_size": 80,
|
| 9 |
+
"T": 700,
|
| 10 |
"channel": 128,
|
| 11 |
"ch_mult": [1, 2, 3, 4],
|
| 12 |
"attn": [2],
|
|
|
|
| 24 |
"test_load_weight": "ckpt_199.pth",
|
| 25 |
"sample_dir": "SampledImgs/",
|
| 26 |
"sampledNoisyImgName": "NoisyNoGuidenceImgs.png",
|
| 27 |
+
"sampledImgName": "SampledNoGuidenceImgs_600.png",
|
| 28 |
"nrow": 8
|
| 29 |
}
|
| 30 |
|