my code like below
output = self.model(input) #predict
#memory check
import gc
import resource
del input, output
gc.collect()
max_mem_used = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
print("{:.2f} MB".format(max_mem_used / 1024))
print always print about 2000M
is there any other memory I need free manually?