search

Ugness / PiCANet-Implementation

Pytorch Implementation of PiCANet: Learning Pixel-wise Contextual Attention for Saliency Detection
MIT License
180 stars 40 forks source link

still have memory leak problem #21

Open lbaodong001 opened 5 years ago

lbaodong001 commented 5 years ago

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