saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-31 23:27:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

change a for iteration (#9678)

Browse source 
Co-authored-by: Min Lin <linmin@microsoft.com>
This commit is contained in:
mindest 2021-11-09 08:33:50 +08:00 committed by GitHub
parent 53afaefe3b
commit c579ebfbc3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
orttraining/orttraining/python/training/ortmodule/_gradient_accumulation_manager.py
View file

@ -55,7 +55,7 @@ class GradientAccumulationManager(object):
forward_outputs (OrtValueVector): List of outputs returned by forward function
"""
if not self.enabled:
return tuple(_utils._ortvalue_to_torch_tensor(forward_output, device) for forward_output in forward_outputs)
return tuple(_utils._ortvalue_to_torch_tensor(forward_outputs[i], device) for i in range(len(forward_outputs)))
if self._update_cache:
for i in range(self._cache_start, len(forward_outputs)):
self.cache.insert(