pytorch-image-models icon indicating copy to clipboard operation
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)...

Results 196 pytorch-image-models issues
Sort by recently updated
recently updated
newest added

**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...

bug

**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...

bug

**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

enhancement

Hi, are the following networks implemented in this repo? `The network in the red curve` ![image](https://user-images.githubusercontent.com/6227348/182567379-582aa15f-4fa1-40e7-8aec-f74d837b9d9f.png) In this paper: [A Battle of Network Structures: An Empirical Study of CNN, Transformer,...

enhancement

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...

enhancement

**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...

enhancement

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...

enhancement

It's Apple's new model for mobiles https://github.com/apple/ml-mobileone

enhancement

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...

enhancement

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...