woshixuhao
commited on
Commit
·
364ae44
1
Parent(s):
6e9bbb5
Update app.py
Browse files
app.py
CHANGED
@@ -1383,7 +1383,7 @@ nn_params = {
|
|
1383 |
'graph_pooling': args.graph_pooling,
|
1384 |
'descriptor_dim': 1827
|
1385 |
}
|
1386 |
-
device =
|
1387 |
model = GINGraphPooling(**nn_params).to(device)
|
1388 |
|
1389 |
|
|
|
1383 |
'graph_pooling': args.graph_pooling,
|
1384 |
'descriptor_dim': 1827
|
1385 |
}
|
1386 |
+
device ='cpu'
|
1387 |
model = GINGraphPooling(**nn_params).to(device)
|
1388 |
|
1389 |
|