Hzzone / Precipitation-Nowcasting

pytorch implemention of trajGRU.
553 stars 168 forks source link

The reason of applying patch size to moving mnist dataset? #7

Closed jhhuang96 closed 2 years ago

jhhuang96 commented 5 years ago

When I read Shi's paper "Convolutional LSTM Network: A Machine Learning:Approach for Precipitation Nowcasting", in his experiments on moving mnist dataset, he shape img data (64*64*1) to (16*16*16). I'm confused about it, is it because of gpu memory and is it appropriate of this way? Thanks!

alexanderhucheerful commented 4 years ago

大佬你好,请问你试过单机多卡训练吗,我使用nn.DataParallel多gpu训练你的模型,但只显示一张卡使用,请问你遇到过这个问题吗,谢谢