Actually having tta in fastai/__init__.py was a mistake. It should have been in fastai/vision/__init__.py. So rather than what @sgugger wrote above, please show us the first few lines of the result of: %prun from fastai.vision import *
Actually having tta in fastai/__init__.py was a mistake. It should have been in fastai/vision/__init__.py. So rather than what @sgugger wrote above, please show us the first few lines of the result of: %prun from fastai.vision import *