Tutorial 11 : Runtime error in train_flow function

[sy-eng]

Thank you for your great tutorials!

I tried tutorial 11 on my laptop (it has no gpu.)
and I got a runtime error in train_flow function.
Its error message said map_location in torch.load should be set.
So I modified
ckpt = torch.load(pretrained_filename)
to
ckpt = torch.load(pretrained_filename, map_location=device).

I guess this modification is good for PCs without gpu.

Thank you.

[phlippe]

Hi @sy-eng, thanks for pointing this out and already preparing the solution! The new commit should now support loading the NFs on CPU. Thanks!

[sy-eng]

Thank you!