Muellerzr fix deepspeed (#23657)
* Fix deepspeed recursion * Better fix
This commit is contained in:
@@ -1708,7 +1708,8 @@ class TrainingArguments:
|
|||||||
"""
|
"""
|
||||||
requires_backends(self, ["torch"])
|
requires_backends(self, ["torch"])
|
||||||
# Make sure `self._n_gpu` is properly setup.
|
# Make sure `self._n_gpu` is properly setup.
|
||||||
_ = self._setup_devices
|
if not hasattr(self, "_n_gpu"):
|
||||||
|
_ = self._setup_devices
|
||||||
return self._n_gpu
|
return self._n_gpu
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
|||||||
Reference in New Issue
Block a user