search

Henry1iu / TNT-Trajectory-Prediction

A Unofficial Pytorch Implementation of TNT: Target-driveN Trajectory Prediction
504 stars 95 forks source link

询问内存大小问题 #11

Closed yizhiweimengxiangfendoudefeifei closed 2 years ago

yizhiweimengxiangfendoudefeifei commented 2 years ago

Hello，您的意思是ram内存必须要大于110G吗？有没有什么办法可以减小内存的使用，因为课题组的电脑内存最大只有34G。顺便问一下，显卡3090，内存34G，可以进行训练吗？

Henry1iu commented 2 years ago

Hi,

你可以增加swap的容量. 另外, 我也有写好ArgoverseInDisk Dataloader你可以换这个进行数据预处理和读取, 但是读取速度较慢. 详情可以参考argoverse_loader_v2.py里的代码.

3090显卡可以进行训练.