MXNET需要改进的地方
2018-01-06 本文已影响22人
有哪个昵称没人用吗
针对使用gluon接口时的典型的需要修改源码的问题
关于Context的问题
Context一般来说可以用 with mxnet.Context(gpu()) 来切换到gpu,cpu同理
但是,这个只在一个文件里可用,如果是import进来(相当于把某个py文件里的代码执行了一次),或者比如加载数据集时启动多个worker,这种时候,default_context的修改往往无法保持,只有在context.py里改默认的context才行