fix: comment

Author: anyangml

deepmd/pt_expt/train/training.py

@@ -324,6 +324,8 @@ def _expand(t: torch.Tensor | None) -> torch.Tensor | None:
         del fparam
     if aparam is not None:
         del aparam
+    if charge_spin is not None:
+        del charge_spin
 
     # make_fx inserts aten.detach.default for saved tensors used in the
     # decomposed autograd.grad backward ops.  These detach nodes break
@@ -1048,7 +1050,7 @@ def _compile_model(self, compile_opts: dict[str, Any]) -> None:
                 ext_coord, ext_atype, nlist_t, mapping, fparam, aparam, charge_spin
             )
             del _warmup_out
-            if DEVICE.type == "cuda" and torch.cuda.is_initialized():
+            if DEVICE.type == "cuda":
                 torch.cuda.synchronize()
 
             wrapper_mod.model[task_key] = _CompiledModel(model, compiled_lower)