alan-turing-institute/affinity-vae

remove batch processing from visualisation functions

marjanfamili opened this issue · 0 comments

where the batch size is chosen to be smaller due to memory issues, if the size of the grids is larger than the batch size chosen, it can result in a memory overload crash. would it be better to call the decoder one by one on individual encodings and append the reconstructions together and form the grid later