我有一些带有大量参数的自动编码器,我想优化其递归版本。例如:
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?