x-evolution 0.1.22__py3-none-any.whl → 0.1.23__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. x_evolution/x_evolution.py +3 -2
  2. {x_evolution-0.1.22.dist-info → x_evolution-0.1.23.dist-info}/METADATA +1 -1
  3. x_evolution-0.1.23.dist-info/RECORD +6 -0
  4. x_evolution-0.1.22.dist-info/RECORD +0 -6
  5. {x_evolution-0.1.22.dist-info → x_evolution-0.1.23.dist-info}/WHEEL +0 -0
  6. {x_evolution-0.1.22.dist-info → x_evolution-0.1.23.dist-info}/licenses/LICENSE +0 -0
x_evolution/x_evolution.py CHANGED
@@ -6,7 +6,6 @@ from pathlib import Path
6
6
  from functools import partial
7
7
 
8
8
  import torch
9
- import torch.distributed as dist
10
9
  from torch import tensor, Tensor, stack, is_tensor, arange, randint
11
10
  from torch.nn import Module, ModuleList, Parameter, ParameterList
12
11
  from torch.optim import SGD, Adam, Optimizer
@@ -126,7 +125,9 @@ class EvoStrategy(Module):
126
125
  # use prepare and run through environment once to sync params
127
126
 
128
127
  wrapped_model = accelerator.prepare(model)
129
- environment(wrapped_model)
128
+
129
+ with torch.no_grad():
130
+ environment(wrapped_model)
130
131
 
131
132
  # get param dictionary
132
133
 
{x_evolution-0.1.22.dist-info → x_evolution-0.1.23.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: x-evolution
3
- Version: 0.1.22
3
+ Version: 0.1.23
4
4
  Summary: x-evolution
5
5
  Project-URL: Homepage, https://pypi.org/project/x-evolution/
6
6
  Project-URL: Repository, https://github.com/lucidrains/x-evolution
x_evolution-0.1.23.dist-info/RECORD ADDED
@@ -0,0 +1,6 @@
1
+ x_evolution/__init__.py,sha256=XcwXJgIMPnCWGfGws3-vKgoR_7IfVslJBtiMvmEeSg0,57
2
+ x_evolution/x_evolution.py,sha256=lvN3ePqD6a5dW1gOv0d1I9yQ4rdv6OuIVvKvXa0yRBM,19126
3
+ x_evolution-0.1.23.dist-info/METADATA,sha256=pYkzZj10t68WjpyFOSMzd8lhRYL9FNGRsHjlcLyP9zw,5728
4
+ x_evolution-0.1.23.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
5
+ x_evolution-0.1.23.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
6
+ x_evolution-0.1.23.dist-info/RECORD,,
x_evolution-0.1.22.dist-info/RECORD DELETED
@@ -1,6 +0,0 @@
1
- x_evolution/__init__.py,sha256=XcwXJgIMPnCWGfGws3-vKgoR_7IfVslJBtiMvmEeSg0,57
2
- x_evolution/x_evolution.py,sha256=ui4HgIcJ_qM0JtL-HzsiwWNF6r5Ybo9c2DG0mWTXo4w,19124
3
- x_evolution-0.1.22.dist-info/METADATA,sha256=FrF0ebgJVX2NrZkjgcH4QHozWfXMhGtTGMpXlRnP-cE,5728
4
- x_evolution-0.1.22.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
5
- x_evolution-0.1.22.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
6
- x_evolution-0.1.22.dist-info/RECORD,,