fixed trainer tr_loss memory leak (#6999)
* fixed trainer tr_loss memory leak * detached returned training loss from computation graph in the Trainer class' training_step() method * Revert "fixed trainer tr_loss memory leak" This reverts commit 47226e4e
Showing
Please register or sign in to comment