问题描述
我最近发现了Google Colab，并上传了我的Pytorch闪电项目，该项目与用于处理音频的训练模型有关。经过一些修改后，我得到了使用TPU训练模型的信息，但是我注意到模型的准确性比我在本地机器上训练的模型差了一半（损耗= 0.02 vs 0.009）。在两种情况下，我都使用相同的超参数。经过一番搜索，我了解到colab使用自己的类型“ bfloat16”而不是float32 precision来提高速度，不确定这是否是问题的一部分。是否有任何明显的原因导致TPU训练的模型准确性降低？如果重要的话，我的模型将使用一维一维卷积层的堆栈。
解决方法
暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！
如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@）