How to reduce memory usage when one large networks called many times
Hi, there, I am a newbie to PyTorch and I am really impressed by its speed and flexibility. Amazing work! But I have one question about the memory usage. Any suggestions would be welcome! Thank you! <SCODE>model = AntoEncoderDecoder()
output = Variable(torch.zeros(256,256))
for i in range(iterations):
    output = model( output )<ECODE>
qianguih
Hi,
albanD
apaszke
qianguih
apaszke
Yes, it is 0.1.10. I guess theano has optimized every computation graph a lot during the function compilation.
qianguih
Maybe. It’s hard to say what’s happening without having the model details.
apaszke