Memory Leak in model.predict

Discussion
1

for k, batch in enumerate(train_dataiter2):

          temp = mod_fc.predict(batch.data[0])

          out = mx.ndarray.cast(data=temp, dtype='float32')

During this operations I noticed memory leaks and don’t know how to solve it.