pytorch-image-models
pytorch-image-models copied to clipboard
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT)...
**Describe the bug** timm0.6.7 version, using timm's resnet50, first convert the model to onnx, and then use tenosrrt's model PTQ quantization function, the quantized model is verified on the val...
**Describe the bug** I am working on quantization of few timm models using Torch FX Graph Mode Quantization. Specifically, I am looking into post training static quantization. For static models...
**Describe the solution you'd like** Add https://github.com/facebookresearch/mvit from [MViTv2: Improved Multiscale Vision Transformers for Classification and Detection](https://arxiv.org/abs/2112.01526) to timm
Hi, are the following networks implemented in this repo? `The network in the red curve`  In this paper: [A Battle of Network Structures: An Empirical Study of CNN, Transformer,...
How can I edit the save-checkpoint inverval for my model? While training with high epochs, old checkpoints are often overwitten by new checkpoints. In my case, the last 10 checkpoints...
**Is your feature request related to a problem? Please describe.** We propose a new basic operation, position aware circular convolution (ParC). Differing from previous convolution operations, the proposed ParC has...
When using timm, the convolutional neural network can be used as the backbone(features_only=True, out_indices=(1, 2, 3, 4)) for dense prediction, but how can the transformer model be used directly as...
It's Apple's new model for mobiles https://github.com/apple/ml-mobileone
We suggest adding DisMax loss to improve OOD detection: https://arxiv.org/abs/2205.05874 It improves the AUROC OOD detection performance of a ResNet18 trained on ImageNet by almost 25% using the hard ImageNet-O...
The branch associated with this PR is for development of `timm bits` -- a significant update and reorganizaiton of timm's training scripts and associated modules. It works w/ TPUs (PyTorch...