Takes a lot of time while providing Imagegridcallback callback in training

[JiteshPshah]

Hi, I used one your function Imagegridcallback for visualization of images on every epoch, I set batch size of 16 and I have 739 image, but when it comes to last batch in every epoch , the process will going to be very slow , means after 736 images the training stops for a while(possibly due to callback of imagegrid ) and then completed and go to next epoch ,
can you optimize imagegridcallbback so it will not take time while generating images on every epoch ?
and if possible can I visualize those epoch images after training ? if yes then how ?