mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 19:33:46 +08:00
fix pytorch 2.0 optimizer no inf check error (#1919)
* fix pytorch 2.0 optimizer no inf check error * Update optimizers.py
This commit is contained in:
1 parent
13e0fce122
commit
445e17a3c6
1 file changed
+2
-1
@@ -127,7 +127,8 @@ class Optimizers:
|
||||
if max_norm is not None:
|
||||
grad_scaler.unscale_(optimizer)
|
||||
torch.nn.utils.clip_grad_norm_(self.parameters[param_group], max_norm)
|
||||
grad_scaler.step(optimizer)
|
||||
if any(any(p.grad is not None for p in g["params"]) for g in optimizer.param_groups):
|
||||
grad_scaler.step(optimizer)
|
||||
|
||||
def optimizer_step_all(self) -> None:
|
||||
"""Run step for all optimizers."""
|
||||
|
||||
Reference in new issue
Block a user