Update logic.py
Browse files
logic.py
CHANGED
|
@@ -392,6 +392,7 @@ def load_data_to_ArangoDB(login):
|
|
| 392 |
|
| 393 |
|
| 394 |
def make_pyg_graph(movie_rec_db):
|
|
|
|
| 395 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 396 |
users = movie_rec_db.collection('Users')
|
| 397 |
movies = movie_rec_db.collection('Movie')
|
|
|
|
| 392 |
|
| 393 |
|
| 394 |
def make_pyg_graph(movie_rec_db):
|
| 395 |
+
global device
|
| 396 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 397 |
users = movie_rec_db.collection('Users')
|
| 398 |
movies = movie_rec_db.collection('Movie')
|