Update graph_decoder/diffusion_utils.py
Browse files
graph_decoder/diffusion_utils.py
CHANGED
|
@@ -5,6 +5,7 @@ import yaml
|
|
| 5 |
import torch
|
| 6 |
import numpy as np
|
| 7 |
from torch.nn import functional as F
|
|
|
|
| 8 |
# from torch_geometric.utils import to_dense_adj, to_dense_batch, remove_self_loops
|
| 9 |
from types import SimpleNamespace
|
| 10 |
|
|
|
|
| 5 |
import torch
|
| 6 |
import numpy as np
|
| 7 |
from torch.nn import functional as F
|
| 8 |
+
## Cause no cuda found issue in zero-gpus
|
| 9 |
# from torch_geometric.utils import to_dense_adj, to_dense_batch, remove_self_loops
|
| 10 |
from types import SimpleNamespace
|
| 11 |
|