fix error reported from training scripts with newer python version: RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation