tnt

tnt copied to clipboard

Add usage logging APIs for the Runner's public entry points

1

Summary: Leverage the feature here: https://pytorch.org/docs/stable/notes/large_scale_deployments.html#api-usage-logging Differential Revision: D38750230

daniellepintz

Add runner.predict

1

Reviewed By: ananthsub Differential Revision: D38692009

daniellepintz

Add runner.evaluate

1

Reviewed By: ananthsub Differential Revision: D38599732

daniellepintz

Add badges to readme

4

Differential Revision: D38480314

ananthsub

add versioning to docs build action

2

Please read through our [contribution guide](https://github.com/pytorch/tnt/blob/main/CONTRIBUTING.md) prior to creating your pull request. Summary: - removes unit tests from doc build (tests still run on PR and commit in the unit_test...

edward-io

Add torchsnapshot checkpoint saver integration

2

Summary: Add a callback to torchTNT which saves checkpoints using torchsnapshot. this relies on the app state mixin defined here: this means users can declare what module/optimizer/etc states they'd like...

ananthsub

Suppress PyRe errors for threaded-PG

2

Summary: Run command ``` pyre --output=json | pyre-upgrade fixme ``` to suppress pyre errors. Differential Revision: D40494695

yhcharles

fix some typos

2

Please read through our [contribution guide](https://github.com/pytorch/tnt/blob/main/CONTRIBUTING.md) prior to creating your pull request. Summary: Test plan: Fixes #{issue number}

edward-io

[draft] stateful dataloader and sampler

2

Please read through our [contribution guide](https://github.com/pytorch/tnt/blob/main/CONTRIBUTING.md) prior to creating your pull request. Summary: Test plan: Fixes #{issue number}

edward-io

Split test_profile_dataloader GPU tests

2

Differential Revision: D55500653

diego-urgell