x-transformers 2.1.6__py3-none-any.whl → 2.1.7__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 (5) hide show
  1. x_transformers/belief_state_wrapper.py +2 -2
  2. {x_transformers-2.1.6.dist-info → x_transformers-2.1.7.dist-info}/METADATA +1 -1
  3. {x_transformers-2.1.6.dist-info → x_transformers-2.1.7.dist-info}/RECORD +5 -5
  4. {x_transformers-2.1.6.dist-info → x_transformers-2.1.7.dist-info}/WHEEL +0 -0
  5. {x_transformers-2.1.6.dist-info → x_transformers-2.1.7.dist-info}/licenses/LICENSE +0 -0
x_transformers/belief_state_wrapper.py CHANGED
@@ -165,8 +165,8 @@ class BeliefStateWrapper(Module):
165
165
  # maybe loss weighting
166
166
 
167
167
  if self.needs_loss_weight:
168
- fb_loss = rearrange(fb_loss, 'b (fb n) -> b fb n')
169
- fb_loss = fb_loss * self.fwd_bwd_loss_weights
168
+ fb_loss = rearrange(fb_loss, 'b (fb n) -> b fb n', fb = 2)
169
+ fb_loss = einx.multiply('b fb n, fb', fb_loss, self.loss_weights)
170
170
  fb_loss = fb_loss.mean()
171
171
 
172
172
  # backwards
{x_transformers-2.1.6.dist-info → x_transformers-2.1.7.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: x-transformers
3
- Version: 2.1.6
3
+ Version: 2.1.7
4
4
  Summary: X-Transformers
5
5
  Project-URL: Homepage, https://pypi.org/project/x-transformers/
6
6
  Project-URL: Repository, https://github.com/lucidrains/x-transformers
{x_transformers-2.1.6.dist-info → x_transformers-2.1.7.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  x_transformers/__init__.py,sha256=NDoiBivau559WQ0FvXG4ssU3Il9aoHmTIUFN_1juz0s,911
2
2
  x_transformers/attend.py,sha256=-5BWWhFsp7tvZTdN91Ay5SqOjyj9uOs-122vFvoO6b4,17253
3
3
  x_transformers/autoregressive_wrapper.py,sha256=reLCno9Z9pchVU79tBF8OMo21LwSZ67KAeB83jqkyAc,10505
4
- x_transformers/belief_state_wrapper.py,sha256=YxHtM5ccdMQMGwzZCNC2Z0S98mugnT4QV6XFk57S05Y,5799
4
+ x_transformers/belief_state_wrapper.py,sha256=WoagJe_cHguWAWzhEAmeDD_TzUsQcMQkgP2Mt88wrAg,5827
5
5
  x_transformers/continuous.py,sha256=p0sCAiH1na236ygwgL1Yyhu36eZBf9cZvoW1JyP_fFE,7073
6
6
  x_transformers/dpo.py,sha256=xt4OuOWhU8pN3OKN2LZAaC2NC8iiEnchqqcrPWVqf0o,3521
7
7
  x_transformers/multi_input.py,sha256=tCh-fTJDj2ib4SMGtsa-AM8MxKzJAQSwqAXOu3HU2mg,9252
@@ -10,7 +10,7 @@ x_transformers/nonautoregressive_wrapper.py,sha256=2NU58hYMgn-4Jzg3mie-mXb0XH_dC
10
10
  x_transformers/x_transformers.py,sha256=-80N4sqUr3sR51Ms4wCfc4jhxnPwf0ApNR4xfIsasfQ,110142
11
11
  x_transformers/xl_autoregressive_wrapper.py,sha256=CvZMJ6A6PA-Y_bQAhnORwjJBSl6Vjq2IdW5KTdk8NI8,4195
12
12
  x_transformers/xval.py,sha256=7S00kCuab4tWQa-vf-z-XfzADjVj48MoFIr7VSIvttg,8575
13
- x_transformers-2.1.6.dist-info/METADATA,sha256=BG9m1_Xpxr_b6swT6fRKIlVhKmUt8G0zkmIxoHV_FT4,87570
14
- x_transformers-2.1.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
15
- x_transformers-2.1.6.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
16
- x_transformers-2.1.6.dist-info/RECORD,,
13
+ x_transformers-2.1.7.dist-info/METADATA,sha256=D7enhdRYdhTddt4kGSS-rSsvtHgtFaLYQH-cVuYxUVM,87570
14
+ x_transformers-2.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
15
+ x_transformers-2.1.7.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
16
+ x_transformers-2.1.7.dist-info/RECORD,,