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__()
|