search

starry-sky6688 / MARL-Algorithms

Implementations of IQL, QMIX, VDN, COMA, QTRAN, MAVEN, CommNet, DyMA-CL, and G2ANet on SMAC, the decentralised micromanagement scenario of StarCraft II
1.46k stars 283 forks source link

参数冻结 #49

Closed kkkclearlove closed 3 years ago

kkkclearlove commented 3 years ago

您好，我想请问下本实验有无保留参数继续训练的方法，我发现在训练过程中经常出现训练崩溃，望您不吝赐教，谢谢！

starry-sky6688 commented 3 years ago

参数里的save_cycle控制训练多少次后保存模型，模型会被保存在参数model_dir下面，然后你加载的时候把model_dir指定好，具体加载模型的代码你可以看一下./policy里面各个算法的init函数里有