search

Hzzone / Precipitation-Nowcasting

pytorch implemention of trajGRU.
547 stars 171 forks source link

trajGRU的配置问题 #1

Closed chaiwentao closed 5 years ago

chaiwentao commented 5 years ago

你好师兄，请问你使用的是pretrained里面的模型吗？我想要装入自己的数据集，并且训练模型，然后输入测试的图片。请问这个具体怎么操作，麻烦了。如果方便，加我QQ：446822646

Hzzone commented 5 years ago

这是我自己训练的模型，加载数据直接看HKOIterator。这个要你自己看代码了。