这个repo是针对https://github.com/quancore/social-lstm.git. 的Social-LSTM项目在Linux下使用GPU进行训练和测试时产生的一系列错误进行了修正,主要改动包括CPU和GPU数据的转换等。改动后可以使用"cuda:0"进行训练和测试。 上述模型已经过训练,训练模型已保存,预测结果已存于result中 -wky 2022/12/27
seuwky1021 / fixed_social-lstm-gpu Goto Github PK
View Code? Open in Web Editor NEWthis repo has some error fix for the original repo https://github.com/quancore/social-lstm.(Errors ablout GPU and CUDA)