# free the memory again
del model
del trainer
torch.cuda.empty_cache()