美文网首页
MXNET需要改进的地方

MXNET需要改进的地方

作者: 有哪个昵称没人用吗 | 来源:发表于2018-01-06 18:16 被阅读22次

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

    关于Context的问题

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

    相关文章

      网友评论

          本文标题:MXNET需要改进的地方

          本文链接:https://www.haomeiwen.com/subject/xenynxtx.html