Terminate the DGS module and delete the torch modules.
DGSModule.terminate()
dgs.models.dgs.dgs.DGSModule.forward
dgs.models.dgs.dgs.DGSModule.validate_params