search

GeoX-Lab / LSTTN

29 stars 4 forks source link

复现问题 #3

Closed QWERTY0205 closed 3 months ago

QWERTY0205 commented 3 months ago

想请问一下这个gpu数量应该怎么调整，没发现有控制这个的参数呢

Juuzoufan commented 3 months ago

想请问一下这个gpu数量应该怎么调整，没发现有控制这个的参数呢

main.py里面的--gpus以及forecasting.py的GPU_NUM

QWERTY0205 commented 3 months ago

哦哦哦 看到了 谢谢您！

Juuzoufan commented 3 months ago

哦哦哦 看到了 谢谢您！

我最近也在学习这篇，可以互相交流一下。