translate-package 0.1.1__py3-none-any.whl → 0.1.2__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. translate_package/models/machine_translation.py +2 -2
  2. {translate_package-0.1.1.dist-info → translate_package-0.1.2.dist-info}/METADATA +1 -1
  3. {translate_package-0.1.1.dist-info → translate_package-0.1.2.dist-info}/RECORD +5 -5
  4. {translate_package-0.1.1.dist-info → translate_package-0.1.2.dist-info}/WHEEL +0 -0
  5. {translate_package-0.1.1.dist-info → translate_package-0.1.2.dist-info}/top_level.txt +0 -0
translate_package/models/machine_translation.py CHANGED
@@ -170,11 +170,11 @@ class MachineTranslationTransformer(pl.LightningModule):
170
170
  self.parameters(), lr=self.lr, weight_decay=self.weight_decay
171
171
  )
172
172
 
173
- if self.model_generation in ["t5", "lstm"]:
173
+ if self.model_generation in ["t5", "lstm", "mt5"]:
174
174
 
175
175
  return [optimizer]
176
176
 
177
- elif self.model_generation in ["bart", "mt5", "nllb"]:
177
+ elif self.model_generation in ["bart", "nllb"]:
178
178
 
179
179
  scheduler = get_linear_schedule_with_warmup(
180
180
  optimizer,
{translate_package-0.1.1.dist-info → translate_package-0.1.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: translate-package
3
- Version: 0.1.1
3
+ Version: 0.1.2
4
4
  Summary: Contain functions and classes to efficiently train a sequence to sequence to translate between two languages.
5
5
  Author: Oumar Kane
6
6
  Author-email: oumar.kane@univ-thies.sn
{translate_package-0.1.1.dist-info → translate_package-0.1.2.dist-info}/RECORD RENAMED
@@ -5,13 +5,13 @@ translate_package/errors/__init__.py,sha256=gu6XjAIghG4lLkYo8x_7_yyLRtK2FIvmC-Wc
5
5
  translate_package/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
6
  translate_package/models/gradient_observation.py,sha256=P91UA5i-RdkK46TqpPOJ54DsUYgTI9cRohgPS1Ch0Lc,294
7
7
  translate_package/models/lstm.py,sha256=OPkvvceowz5JqdGGH4cfPhH23kbP11z-29zIJn5d8ig,3273
8
- translate_package/models/machine_translation.py,sha256=ci_UwkzCF2UG4jsrq1ZERESnFm2sZ25V3TCMY7PPNek,10612
8
+ translate_package/models/machine_translation.py,sha256=GYpxBuscAdBwWixzMzveJPXoJjswC7oWTEwNf5Z8m94,10612
9
9
  translate_package/tokenization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
10
  translate_package/tokenization/load_tokenizer.py,sha256=g8j5pDmimFhwjpeYNkWot0hXMzAqqURbtedcQK-1xYE,1543
11
11
  translate_package/tokenization/train_tokenizer.py,sha256=RkdT5DUx201OBNaswM6m54iqcrmCThd3ITLguQb_zVM,3347
12
12
  translate_package/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
13
13
  translate_package/utils/checkpoint.py,sha256=GqymRvF8_QZgrQq9m79Ppj6Qr7NQm78kDARm3p_chC0,322
14
- translate_package-0.1.1.dist-info/METADATA,sha256=oNiJc5l3kGyz_yxP1DlPin0tftYF6ywdG7Cd3mnGWtk,850
15
- translate_package-0.1.1.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
16
- translate_package-0.1.1.dist-info/top_level.txt,sha256=8e2HIrGAMzoSukqu2q929dOJMV1zGYKI_BAFwl-P7XU,18
17
- translate_package-0.1.1.dist-info/RECORD,,
14
+ translate_package-0.1.2.dist-info/METADATA,sha256=qqmQoxx3LhFG-AtWZsAnNv90hnqUAkdVApQwhkz-jqI,850
15
+ translate_package-0.1.2.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
16
+ translate_package-0.1.2.dist-info/top_level.txt,sha256=8e2HIrGAMzoSukqu2q929dOJMV1zGYKI_BAFwl-P7XU,18
17
+ translate_package-0.1.2.dist-info/RECORD,,