Cannot load pre-trained model

[WLatonov]

An error occurrs when I try to load pre-trained model. The error is catched after the line

GCN_1 = torch::jit::load(firstModelPath);

is executed.

[therateful]

have you solved the problem?

[WLatonov]

have you solved the problem?

Yes. You need to have the same GPU type when you train model and when you load model.

[therateful]

what's the meaning of "GPU type"?Could you please give me an example?

[WLatonov]

NVIDIA Tesla V100, or something else.