search

yuedajiong / super-ai-vision-stereo-world-generate-triposr

20 stars 6 forks source link

Why is my loss "nan"? #7

Open LZUzgw opened 4 hours ago

LZUzgw commented 4 hours ago

I successfully reproduced this code. However, the loss is ‘nan’. What should I do? Is there anyone having the same problem as me?

yuedajiong commented 2 hours ago

训练代码，不是按照作者的网络结构配置的，参数缩小了很多，主要是示意可以正确运行。 你要去找个大GPU，然后简单的把那有限几个参数设置的和作者的一样大，然后训练。

LZUzgw commented 37 minutes ago

训练代码，不是按照作者的网络结构配置的，参数缩小了很多，主要是示意可以正确运行。 你要去找个大GPU，然后简单的把那有限几个参数设置的和作者的一样大，然后训练。

thanks