Spaces:
Runtime error
Runtime error
sunshineatnoon
commited on
Commit
·
cde2253
1
Parent(s):
827b81f
new_model
Browse files- models/week0417/model.py +0 -4
models/week0417/model.py
CHANGED
|
@@ -18,10 +18,6 @@ from swapae.models.networks.stylegan2_layers import ConvLayer
|
|
| 18 |
from torch_geometric.nn import GCNConv
|
| 19 |
from torch_geometric.utils import softmax
|
| 20 |
|
| 21 |
-
import sys
|
| 22 |
-
sys.path.append('models/third_party/cython')
|
| 23 |
-
from connectivity import enforce_connectivity
|
| 24 |
-
|
| 25 |
class GCN(nn.Module):
|
| 26 |
def __init__(self, n_cluster, temperature = 1, hidden_dim = 256):
|
| 27 |
super().__init__()
|
|
|
|
| 18 |
from torch_geometric.nn import GCNConv
|
| 19 |
from torch_geometric.utils import softmax
|
| 20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
class GCN(nn.Module):
|
| 22 |
def __init__(self, n_cluster, temperature = 1, hidden_dim = 256):
|
| 23 |
super().__init__()
|