Merge branch 'main' of github.com:biomed-AI/GraphBepi into main

cquzys · cquzys · commit 31432d228d59 · 2022-12-07T19:52:11.000+08:00
diff --git a/test.py b/test.py
@@ -111,7 +111,7 @@ def seed_everything(seed=2022):
     result_path=f'{args.output}',      # path to save temporary result file of testset
 )
 model.load_state_dict(
-    torch.load(f'./model/BCE_633_GraphBepi/model_-1.ckpt')['state_dict'],
+    torch.load(f'./model/BCE_633_GraphBepi/model_-1.ckpt',map_location='cpu')['state_dict'],
 )
 trainer = pl.Trainer(gpus=[args.gpu],logger=None)
 result = trainer.test(model,test_loader)

Original file line number	Diff line number	Diff line change
`@@ -111,7 +111,7 @@ def seed_everything(seed=2022):`
`111`	`111`	`result_path=f'{args.output}', # path to save temporary result file of testset`
`112`	`112`	`)`
`113`	`113`	`model.load_state_dict(`
`114`		`- torch.load(f'./model/BCE_633_GraphBepi/model_-1.ckpt')['state_dict'],`
	`114`	`+ torch.load(f'./model/BCE_633_GraphBepi/model_-1.ckpt',map_location='cpu')['state_dict'],`
`115`	`115`	`)`
`116`	`116`	`trainer = pl.Trainer(gpus=[args.gpu],logger=None)`
`117`	`117`	`result = trainer.test(model,test_loader)`