search

butchland / fastai_xla_extensions

A Python package to allow fastai to run on TPUs using Pytorch-XLA
https://butchland.github.io/fastai_xla_extensions
Apache License 2.0
36 stars 7 forks source link

default_device no longer sets to overridden default_device method #15

Closed butchland closed 3 years ago

butchland commented 4 years ago

Importing the fastai_xla_extensions.core module overrides the fastai.torch_core.default_device but the next line importing from fastai.vision.all import * still uses the old fastai.torch_core.default_device for some reason.

This wasn't the behavior in the past but something changed recently.

butchland commented 3 years ago

No longer requires setting of default_device in latest version (0.0.4)