Tensorflow - 运行递归自动编码器

时间:2018-06-14 15:03:11

标签: tensorflow memory-management

我有一些带有大量参数的自动编码器,我想优化其递归版本。例如:

def auto_encoder(X):
    ..some layers..
    return new_X 

def recursive_AE(X):
   for i in range(N):
      X=auto_encoder(X,reuse_varaibles=True)
    return X

我注意到当N很大时,tenosrflow在分配时“OOM崩溃”。我的问题是为什么内存使用量增长?或者更好,它是如何增长的?线性? quandratic?

0 个答案:

没有答案