I’m using third party project for training age-gender predicting model.
That’s the problem script 11.
The problem is with every next epoch the memory occupied by the script increased.
The memory usage I observe via nvidia-smi.
I can decrease my batch-size and the training cal last 2 epochs but then cuda runs out of memory again.
I suppose there’s a memory allocation in a wrong place or something like that.
Maybe someone can find such detail in a quick look.
any help would be appreciated.