More GPU -> CPU shifts.

model_def.py

@@ -5,7 +5,8 @@ import torch
 from torch_geometric.nn import SAGEConv, to_hetero, Linear
 from dotenv import load_dotenv
 
-data = torch.load("./PyGdata.pt")
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+data = torch.load("./PyGdata.pt", map_location=device('cpu'))
 
 class GNNEncoder(torch.nn.Module):
     def __init__(self, hidden_channels, out_channels):

visualizer.py

@@ -17,9 +17,9 @@ import yaml
 
 
 
-data = torch.load("./PyGdata.pt")
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+data = torch.load("./PyGdata.pt", map_location=device('cpu'))
 
 
 movies_df = pd.read_csv("./sampled_movie_dataset/movies_metadata.csv")
@@ -65,7 +65,7 @@ class Model(torch.nn.Module):
     
 model = Model(hidden_channels=32).to(device)
 model2 = Model(hidden_channels=32).to(device)
-model.load_state_dict(torch.load("PyGTrainedModelState.pt"))
+model.load_state_dict(torch.load("PyGTrainedModelState.pt"), map_location=device('cpu'))
 model.eval()
 
 total_users = data['user'].num_nodes