这篇文章代码的复现貌似需要TPU,因为导入了jax模块,在实验室的电脑上,这个模块的导入始终有问题,再加上没有TPU,我也不会改代码,所以就放弃了这篇文章代码的复现~真希望有人帮我debug,要是有个程序员男朋友就更好了,哈哈哈哈哈~

更多推荐