errors reslved

siddsax · Nov 17, 2018 · a7214c4 · a7214c4
1 parent 4898c16
commit a7214c4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/utils/data_helpers.py b/utils/data_helpers.py
@@ -67,7 +67,7 @@ def load_data_and_labels(data, M=0, N=0):
        		#y_te = y_te.resize((np.shape(y_te)[0], np.shape(y_tr)[1]))
 	    	Y = sp.csr_matrix((val_idx, (row_idx, col_idx)), shape=(m, N))
     	elif(N < n):
-		Y = sp.csr_matrix((val_idx, (row_idx, col_idx)), shape=(m, N))
+		Y = sp.csr_matrix((val_idx, (row_idx, col_idx)), shape=(m, n))
         	Y = Y[:, :N]
     else:
 	Y = sp.csr_matrix((val_idx, (row_idx, col_idx)), shape=(m, n))

diff --git a/utils/futils.py b/utils/futils.py
@@ -65,8 +65,8 @@ def load_model(model, name, optimizer=None):
         return model
 
 def save_model(model, optimizer, epoch, name):
-    if not os.path.exists('saved_models/' + name):
-        os.makedirs('saved_models/' + name)
+
+
     checkpoint = {
         'state_dict': model.state_dict(),
         'optimizer': optimizer.state_dict(),