Open deneal123 opened 1 week ago
If evaluation is carried out during training, memory will go beyond the boundaries I watched in real time how the reconstruction graph was changing and got a memory overflow :)
Hey @deneal123 thanks for pointing that out! Feel free to open a short PR if you have the fix 🙏🏼 I'll have a look at it ASAP otherwise 👍🏼
If evaluation is carried out during training, memory will go beyond the boundaries I watched in real time how the reconstruction graph was changing and got a memory overflow :)