autogluon.multimodal 1.2.1b20250402__py3-none-any.whl → 1.2.1b20250404__py3-none-any.whl

autogluon/multimodal/optim/lit_module.py

@@ -139,7 +139,7 @@ class LitModule(pl.LightningModule):
             - ia3, ia3_bias, ia3_norm (adds vector that scales activations by learned vectors, in combination with either bit_fit or norm_fit)
             - None (do not use efficient finetuning strategies)
         track_grad_norm
-            Track the p-norm of gradients during training. May be set to ‘inf’ infinity-norm.
+            Track the p-norm of gradients during training. May be set to 'inf'  infinity-norm.
             If using Automatic Mixed Precision (AMP), the gradients will be unscaled before logging them.
 
         """
@@ -485,12 +485,14 @@ class LitModule(pl.LightningModule):
                     len(self.trainer.datamodule.train_dataloader())
                     * self.trainer.max_epochs
                     // accumulate_grad_batches
+                    // self.trainer.num_devices  # Account for distributed training
                 )
                 logger.debug(
                     f"len(trainer.datamodule.train_dataloader()): {len(self.trainer.datamodule.train_dataloader())}"
                 )
                 logger.debug(f"trainer.max_epochs: {self.trainer.max_epochs}")
                 logger.debug(f"accumulate_grad_batches: {accumulate_grad_batches}")
+                logger.debug(f"num_devices: {self.trainer.num_devices}")
         else:
             max_steps = self.trainer.max_steps
 

autogluon/multimodal/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.2.1b20250402"
+__version__ = "1.2.1b20250404"
 __lite__ = False

{autogluon.multimodal-1.2.1b20250402.dist-info → autogluon.multimodal-1.2.1b20250404.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.multimodal
-Version: 1.2.1b20250402
+Version: 1.2.1b20250404
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -45,7 +45,7 @@ Requires-Dist: tqdm<5,>=4.38
 Requires-Dist: boto3<2,>=1.10
 Requires-Dist: torch<2.6,>=2.2
 Requires-Dist: lightning<2.6,>=2.2
-Requires-Dist: transformers[sentencepiece]<5,>=4.38.0
+Requires-Dist: transformers[sentencepiece]<4.50,>=4.38.0
 Requires-Dist: accelerate<1.0,>=0.34.0
 Requires-Dist: requests<3,>=2.30
 Requires-Dist: jsonschema<4.22,>=4.18
@@ -57,9 +57,9 @@ Requires-Dist: scikit-image<0.25.0,>=0.19.1
 Requires-Dist: text-unidecode<1.4,>=1.3
 Requires-Dist: torchmetrics<1.6.2,>=1.2.0
 Requires-Dist: omegaconf<2.3.0,>=2.1.1
-Requires-Dist: autogluon.core[raytune]==1.2.1b20250402
-Requires-Dist: autogluon.features==1.2.1b20250402
-Requires-Dist: autogluon.common==1.2.1b20250402
+Requires-Dist: autogluon.core[raytune]==1.2.1b20250404
+Requires-Dist: autogluon.features==1.2.1b20250404
+Requires-Dist: autogluon.common==1.2.1b20250404
 Requires-Dist: pytorch-metric-learning<2.4,>=1.3.0
 Requires-Dist: nlpaug<1.2.0,>=1.1.10
 Requires-Dist: nltk<3.9,>=3.4.5

{autogluon.multimodal-1.2.1b20250402.dist-info → autogluon.multimodal-1.2.1b20250404.dist-info}/RECORD

@@ -1,8 +1,8 @@
-autogluon.multimodal-1.2.1b20250402-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.multimodal-1.2.1b20250404-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/multimodal/__init__.py,sha256=6WuDt3sMP1QLJs9se-20mmHCsFR-q2ZttPc8s0k1QWA,204
 autogluon/multimodal/constants.py,sha256=eLF3t-447nw3aNrYPh4Y8pycdECGu__wv6TC-amfXw8,9509
 autogluon/multimodal/predictor.py,sha256=4lou5yGysY1O86A-PX8AgsJvjB5bq2eHJ9zB1DyZFew,42847
-autogluon/multimodal/version.py,sha256=fIsRdOTHCNt_TXK_xcHoNj9fqSRNcPrRdSWIVcKlEdw,91
+autogluon/multimodal/version.py,sha256=9N1YcBzrSO8rGscMJ3AkRCIx8w8DfKFrprwJAj1OkRQ,91
 autogluon/multimodal/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/multimodal/cli/prepare_detection_dataset.py,sha256=9NCYmCUMPRWbxxbN_C7YQjMYlrAm8gbwJ3Qd-79JWH4,5218
 autogluon/multimodal/cli/voc2coco.py,sha256=LXVu9RIfOZs_1URrzO-_3Nrvb9uGEgPxCY4-B6m1coc,9605
@@ -103,7 +103,7 @@ autogluon/multimodal/optim/deepspeed.py,sha256=KXIfU7Df5Vm3lCoJ-gT58E93Xdw4kznvH
 autogluon/multimodal/optim/lit_distiller.py,sha256=Nf1gfqPgzxSAETj-hp0bOzpYqQJLK6qM7eOzNjcX0YQ,21533
 autogluon/multimodal/optim/lit_matcher.py,sha256=7OC2PvF_BnV-fTh2mqZdirrlkZULpx8LhSJCB7HrZR8,17664
 autogluon/multimodal/optim/lit_mmdet.py,sha256=rWxmgwRdg9gKnPI3hYTgvkH7nyCoSwg0HANcd3YCqd4,9731
-autogluon/multimodal/optim/lit_module.py,sha256=LJqS2S6mVLc0G9fasB-uTJ-Dqf4WsB5OfsSJF3MSIzw,22624
+autogluon/multimodal/optim/lit_module.py,sha256=Ui3oSAPwoLpioFkSak8qI2q4rxxIswjftPY6jFox6Mk,22778
 autogluon/multimodal/optim/lit_ner.py,sha256=SCmhGRUGz1ELrAG2Tlw1UwxJ3XUcvjvWxJserw-KpTk,8579
 autogluon/multimodal/optim/lit_semantic_seg.py,sha256=9HXNZbLEPriZkJvw23V-MRnapodHnlSFpm9AF3CjWkc,4803
 autogluon/multimodal/optim/utils.py,sha256=h9EHuqJWdjFa40TKHk7PItwt7vujBrGXeK8gvwhkEJU,8871
@@ -153,11 +153,11 @@ autogluon/multimodal/utils/registry.py,sha256=vqvONWweZP44aBo1jCvlqLdL0Agn2kyTK8
 autogluon/multimodal/utils/save.py,sha256=aXZa_iue34dAEfTz7nCaRowktG1emEi5uVXe_tDmHBA,4408
 autogluon/multimodal/utils/strategy.py,sha256=tT9PWh_ZLwNdGFgPsXgZsgKRhpnfBQDjh1mB1_y8G18,833
 autogluon/multimodal/utils/visualizer.py,sha256=qAc4_36r3MBDPq1R1jBeb_gcfzIrsylL1S31sRf3wuw,22562
-autogluon.multimodal-1.2.1b20250402.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.multimodal-1.2.1b20250402.dist-info/METADATA,sha256=IxGCCVtF1_a7jHTwkqfNyuOo7EupQAVZYM1s36xkhyY,13218
-autogluon.multimodal-1.2.1b20250402.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.multimodal-1.2.1b20250402.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.multimodal-1.2.1b20250402.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.multimodal-1.2.1b20250402.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.multimodal-1.2.1b20250402.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.multimodal-1.2.1b20250402.dist-info/RECORD,,
+autogluon.multimodal-1.2.1b20250404.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.multimodal-1.2.1b20250404.dist-info/METADATA,sha256=3sjPIDvc5J-f7b52abgftxc3Of5ljwjpPVBbT1F_01Q,13221
+autogluon.multimodal-1.2.1b20250404.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.multimodal-1.2.1b20250404.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.multimodal-1.2.1b20250404.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.multimodal-1.2.1b20250404.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.multimodal-1.2.1b20250404.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.multimodal-1.2.1b20250404.dist-info/RECORD,,