pytorch-image-models
pytorch-image-models copied to clipboard
huggingface
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT)...
**Is your feature request related to a problem? Please describe.** A clear and concise description of what the problem is. **Describe the solution you'd like** A clear and concise description...
thanks for your great job.It is so useful for me. By the way, do you have a plan to add new model called OmniVec?
**Describe the bug** When running `inference.py --data-dir data --class-map class_map.txt --model efficientnet_b7 --num-classes 8 --checkpoint output/model_best.pth.tar` where `data` is a directory that contains images and no subfolders, the script crashes...
**Is your feature request related to a problem? Please describe.** I’m working on 3D volumetric data (medical OCT volumes) and would like to use timm as a unified model zoo...
Hello, As mentioned in the related issue (https://github.com/huggingface/pytorch-image-models/issues/2622), the CSATv2 model worked only in the HuggingFace environment and failed to load in a standard timm setup. I updated the code,...
**Is your feature request related to a problem? Please describe.** DeepSpeed/ZeRO provides meaningful acceleration and memory reduction for large-scale training. However, `timm`’s training script currently does not include an official...
**Is your feature request related to a problem? Please describe.** I would like to contribute a new high-resolution ImageNet-1K model, **CSATv2**, to the timm model zoo. Right now, CSATv2 is...
Metadata
Owner
Metadata
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT)...