report generation

ck-amrahd · Apr 30, 2020 · cd2f175 · cd2f175
1 parent 3632ff0
commit cd2f175
Show file tree

Hide file tree

Showing 5 changed files with 6,223 additions and 25 deletions.
diff --git a/accuracy.py b/accuracy.py
@@ -10,7 +10,7 @@
 from sklearn.metrics import f1_score
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model_path = 'Results/model.pth'
+model_path = 'Results/model_exp2.pth'
 num_classes = 80
 train_path = '/home/user/Data/coco2014/train2014'
 train_ann_file = '/home/user/Data/coco2014/annotations/instances_train2014.json'

diff --git a/gcn.py b/gcn.py
@@ -10,9 +10,11 @@ def __init__(self, adj_tensor, n_input, n_hidden, n_output):
         self.gc1 = GraphConvolution(n_input, n_hidden)
         self.gc2 = GraphConvolution(n_hidden, n_hidden)
         self.gc3 = GraphConvolution(n_hidden, n_output)
+        # self.gc2 = GraphConvolution(n_hidden, n_output)
 
     def forward(self, inp):
         x = F.relu(self.gc1(inp, self.adj))
         x = F.relu(self.gc2(x, self.adj))
         x = self.gc3(x, self.adj)
+        # x = self.gc2(x, self.adj)
         return x