GPU ran out of memory when rerun "Get latent representation" block

[kenshin20080-lab]

Thank you for your great job, Babygan is so much fun.

But I encountered a problem, when I try to rerun the "Get latent representation" block with a new father and mother's pictures, the following messgage showed up and the program terminated later:

Allocator (GPU_0_bfc) ran out of memory trying to allocate 2.26GiB with freed_by_count=0. The caller indicates that this is not a failure, but may mean that there could be performance gains if more memory were available.

Do you have any ideas? Thanks
Sorry for my English