mx.kvstore.create()的取值

kv = mx.kvstore.create()分布式

kvstore主要是解决你的梯度更新是在cpu进行仍是gpu进行

kv = mx.kvstore.create("device") ,表示在GPU上计算梯度和更新权重.net

kv=mx.kvstore.create("local"),表示在cpu上更新blog

kv=mx.kvstore.create("dist_device_sync"),表示分布式训练get

参考文献:cpu

https://blog.csdn.net/u014380165/article/details/73928755di

相关文章
相关标签/搜索