AEI training weights loading

taotaonice / FaceShifter

Try to reproduce FaceShifter

250 stars 62 forks source link

AEI training weights loading #15

Closed ocastan closed 4 years ago

ocastan commented 4 years ago

Can you please explain why you are loading D and G weights on the cpu with a model on gpu ? Thanks in advance

ocastan commented 4 years ago

I've tested and received an error about a seek interface missing. Looking at examples I have noticed torch.load is assigned to a python variable before invoking load_state_dict. I guess the GPU object is missing a seek interface and I can't directly upload to GPU.