0719-2318
Browse files- diffusion.py +6 -6
- quantify_results.ipynb +0 -0
diffusion.py
CHANGED
|
@@ -231,7 +231,7 @@ class TrainConfig:
|
|
| 231 |
push_to_hub = True
|
| 232 |
hub_model_id = "Xsmos/ml21cm"
|
| 233 |
hub_private_repo = False
|
| 234 |
-
dataset_name = "/storage/home/hcoda1/3/bxia34/scratch/LEN128-DIM64-CUB8.h5"
|
| 235 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 236 |
world_size = torch.cuda.device_count()
|
| 237 |
# repeat = 2
|
|
@@ -583,7 +583,7 @@ class DDPM21CM:
|
|
| 583 |
return x_last
|
| 584 |
# %%
|
| 585 |
|
| 586 |
-
num_train_image_list = [
|
| 587 |
|
| 588 |
def train(rank, world_size):
|
| 589 |
config = TrainConfig()
|
|
@@ -689,10 +689,10 @@ if __name__ == "__main__":
|
|
| 689 |
|
| 690 |
params_pairs = [
|
| 691 |
(4.4, 131.341),
|
| 692 |
-
(5.6, 19.037),
|
| 693 |
-
(4.699, 30),
|
| 694 |
-
(5.477, 200),
|
| 695 |
-
(4.8, 131.341),
|
| 696 |
]
|
| 697 |
for params in params_pairs:
|
| 698 |
print(f" sampling for {params}, world_size = {world_size} ".center(100,'-'))
|
|
|
|
| 231 |
push_to_hub = True
|
| 232 |
hub_model_id = "Xsmos/ml21cm"
|
| 233 |
hub_private_repo = False
|
| 234 |
+
dataset_name = "/storage/home/hcoda1/3/bxia34/scratch/LEN128-DIM64-CUB8-4.4-131.341.h5"
|
| 235 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 236 |
world_size = torch.cuda.device_count()
|
| 237 |
# repeat = 2
|
|
|
|
| 583 |
return x_last
|
| 584 |
# %%
|
| 585 |
|
| 586 |
+
num_train_image_list = [800]
|
| 587 |
|
| 588 |
def train(rank, world_size):
|
| 589 |
config = TrainConfig()
|
|
|
|
| 689 |
|
| 690 |
params_pairs = [
|
| 691 |
(4.4, 131.341),
|
| 692 |
+
# (5.6, 19.037),
|
| 693 |
+
# (4.699, 30),
|
| 694 |
+
# (5.477, 200),
|
| 695 |
+
# (4.8, 131.341),
|
| 696 |
]
|
| 697 |
for params in params_pairs:
|
| 698 |
print(f" sampling for {params}, world_size = {world_size} ".center(100,'-'))
|
quantify_results.ipynb
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|