torchtnt

View on PyPIReverse Dependencies (0)

0.2.4 torchtnt-0.2.4-py3-none-any.whl

Wheel Details

Project: torchtnt
Version: 0.2.4
Filename: torchtnt-0.2.4-py3-none-any.whl
Download: [link]
Size: 163459
MD5: a3b2509a77e5a24748e6c01554446c2e
SHA256: c9b738232090c5e3453f202b1feb45e23d5cd2f23a04ca8b53e9b28ad3755ddb
Uploaded: 2024-05-22 16:19:12 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: torchtnt
Version: 0.2.4
Summary: A lightweight library for PyTorch training tools and utilities
Author: PyTorch
Author-Email: daniellepintz[at]fb.com
Home-Page: https://github.com/pytorch/tnt
License: BSD-3
Keywords: pytorch,torch,training,tools,utilities
Classifier: Development Status :: 2 - Pre-Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: BSD License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.7
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.7
Requires-Dist: torch
Requires-Dist: numpy
Requires-Dist: fsspec
Requires-Dist: tensorboard
Requires-Dist: packaging
Requires-Dist: psutil
Requires-Dist: pyre-extensions
Requires-Dist: typing-extensions
Requires-Dist: setuptools
Requires-Dist: tqdm
Requires-Dist: tabulate
Requires-Dist: parameterized; extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: pytest-cov; extra == "dev"
Requires-Dist: torchsnapshot-nightly; extra == "dev"
Requires-Dist: pyre-check; extra == "dev"
Requires-Dist: torchvision (==0.15.2); extra == "dev"
Provides-Extra: dev
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 1778 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.43.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
torchtnt/__init__.py sha256=D-Ymioenry1C6mBiJGrarSEdT3XQj0NVpCQ6gybaG34 246
torchtnt/py.typed sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
torchtnt/framework/__init__.py sha256=J_wM1PZFQqzQo8yVUQ8XTliTKe3ppbIenGjg-58l5II 860
torchtnt/framework/_callback_handler.py sha256=V_RCVH2-uhic2m__p08y6-xZ7AZp45_DrDzWZQbOkrA 8872
torchtnt/framework/_loop_utils.py sha256=K0fapWo_ndmd5eeoLei7NVL0pUE68AuQhmXbrFLsTaU 3414
torchtnt/framework/_test_utils.py sha256=eX4gIUQ0Vimi97Vwes4MDH_D7kXY_0yYPHP_LsvxeLg 6009
torchtnt/framework/_unit_utils.py sha256=ZG7CA9xS8OuqP7qUCnaVBKsKLLOp-YuZqyH2jJysmlQ 1979
torchtnt/framework/auto_unit.py sha256=NeTxFcVVAKBkofepgngUyFUqsugVNbAmydUKmOQDytQ 36138
torchtnt/framework/callback.py sha256=HRfBckwBdLBVwpouBBvz3xc4n3Qyw58g7129pGqx1Io 5829
torchtnt/framework/evaluate.py sha256=QV9vlL_WjfnNww3ybi55B_ChTNN_aDa5vNzqA92SFqQ 7014
torchtnt/framework/fit.py sha256=VlUp41HJkRkWyoUVcfhLc4L41lCFA3lJcPPuUsDg2fo 6509
torchtnt/framework/predict.py sha256=8iZiGatbPZqu6P-mi9AdtJBroVD771O2qM3ZErvHRr4 7557
torchtnt/framework/state.py sha256=hVbNORS-StZqLkcr-sW7sfo3D4MdODKQRfYiMqRQfYs 7900
torchtnt/framework/train.py sha256=MFXocDxkVF9mLNxuvHd529YgN7pkm-kFulHiS18B22M 10234
torchtnt/framework/unit.py sha256=MHyETtdMyDZmjnhDXN142zIJQsun7pCC3XMMxFv1cMA 22471
torchtnt/framework/utils.py sha256=dihZsoxu-80w333MWTgvGxcz1p6kc8_U8dbVSfAcNOg 1178
torchtnt/framework/callbacks/__init__.py sha256=l0iSf1-ewg0Xb62XpPCYNquGMjeUwEfRi18igBe4adQ 1886
torchtnt/framework/callbacks/_checkpoint_utils.py sha256=EouIo3Gg5xgoWO_ix-TavVpYTM1riMiZ0bGB8SPkH_s 3314
torchtnt/framework/callbacks/base_checkpointer.py sha256=UxWRW2U2L-NjIu0Acur9aJSFLw2kFyP8sdjKwEwlf1M 19248
torchtnt/framework/callbacks/base_csv_writer.py sha256=XNfSle_BSE3nvyfxYgReakv0gRvdUI8eXxBCLblrYaM 3364
torchtnt/framework/callbacks/checkpointer_types.py sha256=sZe4XJUCRUvLDw2PIQKQuKJ6WjNT2wa75VBKtF1Gylo 1774
torchtnt/framework/callbacks/dcp_saver.py sha256=Yl21fdtWfKaHU2XAZVgs_2zGFsmasNIv7RajCgT8zf0 15782
torchtnt/framework/callbacks/early_stopping.py sha256=_GRtosR0D2u5JmJjQ3Z_oZVvlFXDj-e1CanR1MIucDo 3017
torchtnt/framework/callbacks/empty_cuda_cache.py sha256=1JMosrYeY8K1qOM1TFCPHRIvjFh831L6WJbMvx2duww 2764
torchtnt/framework/callbacks/garbage_collector.py sha256=NQ7TcqqIGj0SHG8_-Ms9MRi82IMXtyxopUc_ZG1V_18 4332
torchtnt/framework/callbacks/iteration_time_logger.py sha256=W2vWSs7CY_pfsjj0_YBAKhkPTzI2w0Ii62ugBgVZXaQ 5604
torchtnt/framework/callbacks/lambda_callback.py sha256=jfWsiQeUFnmaGrR5-nuLEc-otd3OJTTAjM64lloiAHg 5123
torchtnt/framework/callbacks/learning_rate_monitor.py sha256=M1A8cWoPOLGcWbmwMbLW1P0Cmv5_UGQzyU8_dyOsiTU 3181
torchtnt/framework/callbacks/memory_snapshot.py sha256=HMlK2Iu7ecq6FT3JSHR0kz_p8SUye7p8dThiMkIg3ps 1695
torchtnt/framework/callbacks/module_summary.py sha256=Jde7FQgb0GYOi-aWHNa68VNtm7_Pvgr365Q8TVfry2I 3192
torchtnt/framework/callbacks/progress_reporter.py sha256=fQ2pJnpIIMA8Wt0In3C5wepuj_kFqYQQDTVpKjdCxls 4709
torchtnt/framework/callbacks/pytorch_profiler.py sha256=TodMB2rMBdCcvY4CkP-4bQ75Kr_1Yq51Y62HnUG4myY 2013
torchtnt/framework/callbacks/slow_rank_detector.py sha256=QM0hXxgHIGmaNdznMng5DWXsh9tRXhkvpZkQZBNUchw 5352
torchtnt/framework/callbacks/system_resources_monitor.py sha256=Xqlol_Ayr6onv6Dhsb1bGE7qs7VKAmZVeyApvuU7mGc 3402
torchtnt/framework/callbacks/tensorboard_parameter_monitor.py sha256=uFIWE6y4MEbjloCg3UdM1J3FWqNT_Hs07TZUPcopmrI 1875
torchtnt/framework/callbacks/throughput_logger.py sha256=0xzLvWzEj1ZOTayie1Pgk1_LRBrYKFgImJdLhqHsXWU 8965
torchtnt/framework/callbacks/time_limit_interrupter.py sha256=wFyox23PsimaG-qS3dkoH9xOi1W0_S2lbjB8XodLeTQ 5854
torchtnt/framework/callbacks/time_wait_for_batch_logger.py sha256=YZYEUKe8Unbf1cC6es7c9jcvxbDDHyvuyvF_WmTawNw 3411
torchtnt/framework/callbacks/torch_compile.py sha256=_8P_zTWo17Pljx6lXeJieJioCM8EviTkxQxE46xK8s8 1488
torchtnt/framework/callbacks/torchsnapshot_saver.py sha256=tzHfJfXYCLlsPsJdm2fkvZfF9VXxuJLv13nULRzKEmc 17312
torchtnt/framework/callbacks/tqdm_progress_bar.py sha256=xCDoc9-3yfbCzMcs7nONIbRe9eIXskopBSyxHthZ8a0 5257
torchtnt/framework/callbacks/train_progress_monitor.py sha256=CKp0saYp9thTjGQPkfyebkpW8yQVhNIK5sg9T40ZbhI 1826
torchtnt/utils/__init__.py sha256=Qs4yCc6G3l1k1LHACQnWQPDSFDw_2zsLugDtASbiiF4 4285
torchtnt/utils/checkpoint.py sha256=A3OlX9eovxtf9-b1P3GMglPXEyGhRlrL4qYX7hLUAqM 29240
torchtnt/utils/device.py sha256=-mOBcw6czSVRoOKsNkZoFWZgqKiawcclwMnp4HH5UVA 11546
torchtnt/utils/distributed.py sha256=bsrk_INaBiaPoYqfa6sghhXSsy2wsqZMrrZ0AO42vdU 23329
torchtnt/utils/early_stop_checker.py sha256=SC9a1TldzOWzENV63lwvh_VPCh_8up7de2_dfcFMSZE 9912
torchtnt/utils/env.py sha256=N5-8bAeYe5YsIDmdQCWsXqyVI5PdIaedLzElIv_KeNA 6406
torchtnt/utils/flops.py sha256=uEmDmXpi_M_V4JEn9UX4yFwes4AL5wtIgyAlNKmoD8Y 11324
torchtnt/utils/fsdp_utils.py sha256=6HdQ0520sBOQbndRi6yRPVLI5P2hlbzzJbfCbqtOs4k 4717
torchtnt/utils/fsspec.py sha256=PKBW2nkCFmbE-Ob-O5zJz3h8_u99ZUhJOytEz8BSY2I 530
torchtnt/utils/lr_scheduler.py sha256=MV4Mszr-ZtivigjyWlDMHl9d42db_yY0n8U2QYrB5XY 522
torchtnt/utils/memory.py sha256=dMehJB36_KSaba7rZ7Nct86nMg3cE0wwxQou2qisGm0 4570
torchtnt/utils/memory_snapshot_profiler.py sha256=7n4_-UV4HPPLkVjEYbt1JQtJnzhL3F-6HOOlVmxygg4 6532
torchtnt/utils/misc.py sha256=wpjagqCkOk60tFJSs8BI_3SpLGyJ1Wfgtlcc4h0etvc 2553
torchtnt/utils/module_summary.py sha256=mlbQ1XXgxCr-abKaw552UVSX3mO9ayL3qihXIyuGJRI 27230
torchtnt/utils/oom.py sha256=jgSrKlK50NoF7SyeYNkRMEy2hmm7ualdR5op71hptf8 4975
torchtnt/utils/optimizer.py sha256=ChBfahheuS9nGec49EQh5uzqynOdjBJbFriZe1bduP8 2251
torchtnt/utils/precision.py sha256=MskP0nu7FXi6m-bsB8GbFHhTqwC7u2rVDi6xq0OHy0U 1903
torchtnt/utils/prepare_module.py sha256=4gcV0k7-4U49JPWPLACXsLimL8Cup5NUf3DLNSPaFgw 14932
torchtnt/utils/progress.py sha256=nXPAV2GZbl_ocK5z7DzMiqJ3oklH-vJeOPGB2cuaQMY 5744
torchtnt/utils/rank_zero_log.py sha256=CKdzc9JrsUyaSdiSaZB5u7WNNzFrbl6Xh_jVhbuqn1w 2404
torchtnt/utils/stateful.py sha256=AhvMWF1hQhkJ8jh4KuWu3OyG3CGGZUXH5mURI4DlDUA 1725
torchtnt/utils/swa.py sha256=Q-M6CgbwSjkYe0n1y9B23JmvmMiFTf-SyyTklIR_pMg 4215
torchtnt/utils/test_utils.py sha256=kah6PhEd1ZCsLRsXzpCktrTaGPc3525hiVie_qPxZlk 2986
torchtnt/utils/timer.py sha256=qVeOqI-SjhokyLaYDcdZesm0f13qUJVlSQ5rir1Cwvg 15408
torchtnt/utils/tqdm.py sha256=0-oGaT9JXQg5K5_9HTAQdccD4pxlLhpNElg8SNytC-U 2474
torchtnt/utils/version.py sha256=6OFKL3E6RPHYYrffUejJxs4xX7OBAVJbNKctCsyX2gQ 1472
torchtnt/utils/data/__init__.py sha256=FlHk8rbhC5nNX7rwoKfZ2BVPCa_B4kCwpAP5Sa7zeF4 965
torchtnt/utils/data/data_prefetcher.py sha256=_1xsjlzpuZl1rSWdve3P2qJA3ueEUOdOQz45AOvAGSA 3851
torchtnt/utils/data/iterators.py sha256=jNkVv0Ud7edDo5PU38QF4yswmMf14VeDsjAhirQ9t6U 21149
torchtnt/utils/data/multi_dataloader.py sha256=wmb1nKAawfICOmeG24BYxSzPSnspkE_fuMNb4tzHV6U 5912
torchtnt/utils/data/profile_dataloader.py sha256=SRhhzflgFAM92dtksjDhLg5DnD3zg_oxJNWjRZSPc3U 2213
torchtnt/utils/data/synthetic_data.py sha256=ej2yyX9pnwJwvDyHLxiUt8AqqaFDNGLdaqrbqLbm1Bs 3430
torchtnt/utils/loggers/__init__.py sha256=cDJ1MHetVIdm2qTQ9818kqKe4wGdiOlZZ_PI0F42X4Y 692
torchtnt/utils/loggers/csv.py sha256=Tazb5hNHmsUt0Jp0DS5M2_76cXgrqIQYwXh4M6NQ53E 2330
torchtnt/utils/loggers/file.py sha256=nspdmRD_7qhTVfE3Oe-pCSwl_u0CuRPCQG0p7r6z_2U 2880
torchtnt/utils/loggers/in_memory.py sha256=GlEnf-2nos43f0sjMQSHRYICbla3IIvblm-64BVBwuc 2105
torchtnt/utils/loggers/json.py sha256=niGNkAYNOXLjzOjprlSk5VLM_hCNitcFzuVfCgjXx7I 1370
torchtnt/utils/loggers/logger.py sha256=gaQw9TUjC66mkEjSG_dF8b-Oxj3Nn3a04Ajt1vm3BNw 1268
torchtnt/utils/loggers/stdout.py sha256=XKTe2gJMej1Wg50gFTc4KX60yYcYFcX5GT8WynJ0FUE 2671
torchtnt/utils/loggers/tensorboard.py sha256=kkpmkbM0R4977LZvVI_cRLHsRluLtzMjl-Sdbqky0hw 7358
torchtnt/utils/loggers/utils.py sha256=oLkUqP1vM4KNJma1rgBXMpLSBMQEdAZqZw0ZrdQ2X-w 947
torchtnt-0.2.4.dist-info/LICENSE sha256=-fVJK02wFtUOab8ZayV3gekkl3SMq9Er87ve5F-ibxI 1533
torchtnt-0.2.4.dist-info/METADATA sha256=QAg-8oIU3fMTNlDKPfbSnNGqcj2VtXWpMPd-B_d2vhQ 3077
torchtnt-0.2.4.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
torchtnt-0.2.4.dist-info/top_level.txt sha256=kujqausXz885MjZc3yaQcS-skRhX8Xo03wpUOpYvRBI 9
torchtnt-0.2.4.dist-info/zip-safe sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs 1
torchtnt-0.2.4.dist-info/RECORD

top_level.txt

torchtnt

zip-safe