Anmol Sood

Out of Memory during Training

1

After a few iterations, there is an out of memory error while training (with the default settings). The memory used by the model is increasing after every iteration quite rapidly....

Hi, Thanks for your work on implementing this model in PyTorch. Just wanted to know if this implementation managed to reproduce the results mentioned in the paper?