Run_backward: expected dtype Float but got dtype Long

Are you seeing this issue using FastAI as well or plain PyTorch?