Skip to content

v0.3.0 - contrib QAT example, additional converters

Choose a tag to compare

View all tags
@jaybdub jaybdub released this 15 Jul 19:53
· 310 commits to master since this release
v0.3.0
c50039a
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.

This version includes the introduction of the Quantization Aware Training workflow in torch2trt.contrib (thanks to @SrivastavaKshitij).
It also contains various converters added since the previous release. Please see the notes below.

Added

  • Added converter for torch.nn.functional.adaptive_avg_pool3d
  • Added converter for torch.nn.functional.adaptive_max_pool3d
  • Added converter for torch.maxpool3d and torch.nn.functional.max_pool3d
  • Added Quantization Aware Training (QAT) workflow to contrib
  • Added converter for torch.roll
  • Added converter for torch.nn.functional.layer_norm
  • Added converter for torch.nn.functional.gelu
  • Added converter for torch.nn.functional.linear
  • Added converter for torch.nn.functional.silu
Assets 2
Loading