为pytorch实现内存monger方法

时间:2017-09-25 16:46:53

标签: pytorch

我想实现本文Training Deep Nets with Sublinear Memory Cost中描述的内存monger方法。可以找到为mxnet框架实现该方法的代码here

我的问题是,这个内存monger方法适用于像Theano和Tensorflow中的符号图,但是pytorch中的图是动态的,所以这个方法可以应用于pytorch吗?

0 个答案:

没有答案