Skip to content

fix gnn inference map location

Alina Lazar requested to merge alina_fix_gnn_infer into dev

Add map_location=torch.device("cpu") to torch.load in the inference stage.

Merge request reports