mx.kvstore.create()的取值

kv = mx.kvstore.create()

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

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

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

kv=mx.kvstore.create("dist_device_sync"),表示分佈式訓練

參考文獻：

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