MXNET需要改进的地方

2018-01-06  本文已影响22人  有哪个昵称没人用吗

针对使用gluon接口时的典型的需要修改源码的问题

关于Context的问题

Context一般来说可以用 with mxnet.Context(gpu()) 来切换到gpu,cpu同理
但是,这个只在一个文件里可用,如果是import进来(相当于把某个py文件里的代码执行了一次),或者比如加载数据集时启动多个worker,这种时候,default_context的修改往往无法保持,只有在context.py里改默认的context才行

上一篇下一篇

猜你喜欢

热点阅读