Помочь устранить утечку оперативной памяти в приложении на Python

Есть приложение на Python (нейросеть на Pytorch). При его использовании, при каждом прогоне через нейросеть (на вход подается изображение, на выходе получается результат) используемая оперативная память возрастает примерно на 10 мб, что в итоге приводит к необходимости перезагружать приложение. Нужно посмотреть код и найти, где утечка, и как её устранить. Нейросеть небольшая, примерно 200 строчек кода, в основном слои.