New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

xla_fit_one_cycle fails with an error when using fastai version 2.3.1 #35

Open

butchland opened this issue May 9, 2021 · 0 comments

Owner

butchland commented May 9, 2021 •

edited

Loading

When running an xla fit method, it triggers an error:

Exception in device=TPU:0: Input type (torch.FloatTensor) and weight type (XLAFloatType) should be the same

Note: downgrading to 2.3.0 is a temporary workaround while this is being fixed
Works with fastai==2.3.0
See this gist for a error sample

The text was updated successfully, but these errors were encountered:

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment