FurkanNar
/

Spatial_Context_Networks

Text Classification

Model card Files Files and versions

FurkanNar commited on Feb 26

Commit

f17d188

·

verified ·

1 Parent(s): 85c6bd9

Update train.py

Files changed (1) hide show

train.py +1 -1

train.py CHANGED Viewed

@@ -11,7 +11,7 @@ import torch.nn as nn
 import torch.optim as optim
 from torch.utils.data import DataLoader, TensorDataset
-from spatial_context_networks import SpatialContextNetwork
 def make_synthetic_dataset(n_samples=256, input_dim=10, output_dim=4, seed=42):

 import torch.optim as optim
 from torch.utils.data import DataLoader, TensorDataset
+from model import SpatialContextNetwork
 def make_synthetic_dataset(n_samples=256, input_dim=10, output_dim=4, seed=42):