When loading the .pth weight file, where are they loaded when using cpu and cuda respectively?

After loading the weight file, if it is loaded to the cpu version, then the weight file will exist in the process memory, and loaded to the cuda version, should the weight file be loaded into the process memory or the cuda video memory?

1 Like