There is still have the memory leak problem when running train.py downloaded from https://github.com/Ugness/PiCANet-Implementation.