fast-bert icon indicating copy to clipboard operation
fast-bert copied to clipboard

Published 20 hours ago • verified publisher icon utterworks

Is it possible to run with TPU on Colab?

Open leejason opened this issue 5 years ago • 2 comments

It would be great if TPU is possible.

leejason avatar Jan 25 '20 03:01 leejason

Not yet. Haven’t evaluated this yet but is on my radar.

kaushaltrivedi avatar Feb 04 '20 14:02 kaushaltrivedi

seems to be the same case even with Google Colab Pro and GPU activated?

Warning: multi_tensor_applier fused unscale kernel is unavailable, possibly because apex was installed without --cuda_ext --cpp_ext. Using Python fallback. Original ImportError was: ModuleNotFoundError("No module named 'amp_C'",)

kirilzilla avatar Feb 27 '20 20:02 kirilzilla