MXNET需要改进的地方

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

关于Context的问题

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

你可能感兴趣的:(MXNET需要改进的地方)