after retraining a cnn_learner, export()'ing it, and reloading it with load_learner(), when I then then run reloaded_model.get_preds(…), I get the following error:
“RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same”
hm so if I comment out the .to_fp16() then the script above runs fine with no errors. And I think from this issue it looks like this is probably a bug? https://github.com/NVIDIA/apex/issues/168
…but maybe I’m just missing the correct way to export/reload a model trained with fp16 precision…