woshixuhao commited on
Commit
364ae44
·
1 Parent(s): 6e9bbb5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1383,7 +1383,7 @@ nn_params = {
1383
  'graph_pooling': args.graph_pooling,
1384
  'descriptor_dim': 1827
1385
  }
1386
- device = args.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