Skip to content

PyTorch Anchor-based PlainNet Model (ABPN)
Compare
Choose a tag to compare
@quic-bharathr quic-bharathr released this 26 Apr 23:00
· 22 commits to develop since this release
abpn-checkpoint-pytorch
6a8efcb

The release provides the model checkpoint tarballs for different variations of the PyTorch-based Anchor-based PlainNet Model (ABPN). Each model tarball corresponds to a given num_channels and scaling_factor (release_abpn_<num_channels>_<scaling_factor>x.tar.gz). Each tarball contains the following:

  • checkpoint_float32.pth.tar - full-precision model with the highest validation accuracy on the DIV2k dataset
  • checkpoint_int8.pth - quantized model with the highest validation accuracy obtained with Quantization-aware Training using AIMET
  • checkpoint_int8.encodings - Encodings for the quantized models
Assets 8
Loading