Results 2 comments of CodeCat

多GPU训练时保存的模型会在每一个key前面加上module,你可以在载入时,通过切片的方式将其去掉,例如:key[7:],从索引7开始。如果只有一张卡,可以不用采用多GPU训练。

> Reference 没有使用