0725-1814
Browse files- diffusion.py +6 -0
diffusion.py
CHANGED
|
@@ -27,6 +27,11 @@
|
|
| 27 |
# 4 GPU, batch_size = 10, num_image = 3200,
|
| 28 |
|
| 29 |
# %%
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
from dataclasses import dataclass
|
| 31 |
import h5py
|
| 32 |
import torch
|
|
@@ -67,6 +72,7 @@ import torch.distributed as dist
|
|
| 67 |
|
| 68 |
import argparse
|
| 69 |
|
|
|
|
| 70 |
# %%
|
| 71 |
def ddp_setup(rank: int, world_size: int, master_addr, master_port):
|
| 72 |
"""
|
|
|
|
| 27 |
# 4 GPU, batch_size = 10, num_image = 3200,
|
| 28 |
|
| 29 |
# %%
|
| 30 |
+
import logging
|
| 31 |
+
#logging.getLogger("torch").setLevel(logging.ERROR)
|
| 32 |
+
import warnings
|
| 33 |
+
warnings.filterwarnings("ignore", message=r"^Detected kernel version")
|
| 34 |
+
|
| 35 |
from dataclasses import dataclass
|
| 36 |
import h5py
|
| 37 |
import torch
|
|
|
|
| 72 |
|
| 73 |
import argparse
|
| 74 |
|
| 75 |
+
|
| 76 |
# %%
|
| 77 |
def ddp_setup(rank: int, world_size: int, master_addr, master_port):
|
| 78 |
"""
|