Merge branch 'refactor' of github.com:BaguaSys/bagua into refactor

BaguaSys · Jul 1, 2021 · 1cfccfe · 1cfccfe
2 parents 1628a88 + cde905f
commit 1cfccfe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bagua/torch_api/utils.py b/bagua/torch_api/utils.py
@@ -119,7 +119,7 @@ def set_storage(param, weight_storage, grad_storage, storage_offset):
     for params in params_list:
         for p in params:
             # copy data
-            flatten_weights_tensor[offset:offset + p.numel()] = p.data.reshape(-1)
+            flatten_weights_tensor[offset : offset + p.numel()] = p.data.reshape(-1)
 
             if p.grad is not None:
                 flatten_grads_tensor[offset : offset + p.numel()] = p.grad.data.reshape(