记一下多GPU训练LSTM
使用多GPU需要在to.(device)前面加model = nn.Dataparallel(model)有错误lstm只有一个输入x,原代码中多了一个self.hidden_vect_1,去掉后如下output, hidden = self.lstm(x)能跑但是有warning需要在lstm前面加一行self.lstm.flatten_parameters()这样就没问题了总结在to(device)前面加nn.Dataparallelmodel = nn.Datap
原创
2020-09-16 10:24:35 ·
1186 阅读 ·
0 评论