trainers.SGMCForceMatching.move_to_device

Contents

`trainers.SGMCForceMatching.move_to_device`#

SGMCForceMatching.move_to_device()#: Move all attributes that are expected to be on device to device to avoid TracerExceptions after loading trainers from disk, i.e. loading numpy rather than device arrays.