pytorch-image-models

pytorch-image-models copied to clipboard

Handle and warn on incompatible kwargs during model creation to enhance the robustness and flexibility of create_model

3

# **Motivation** When using `timm` as a backbone library, users often switch between different model architectures, such as CNNs and Vision Transformers. These models frequently have different `__init__` signatures. For...

developer0hye

## Fix infinite recursion risk in ImageDataset error handling ### Changes - Added maximum retry limit to prevent infinite recursion when loading corrupted/missing files - Improved error messages and logging...

mohiuddin-khan-shiam

I use validate.py to validate the performance of the model on a new dataset, but I need these as indicators. Is it possible to increase these indicators?

chenyanting1

Given the popularity of Agglomerative Vision Foundation Models such as RADIO, RADIOv2.5. It would be great if they can be added into TIMM. https://github.com/NVlabs/RADIO

RaptorMai

Refactor DPN model

2

Refactor the DPN model in `timm` style with support for `forward_intermediates`, and map the checkpoint weights to ensure output is the same.

brianhou0208

[FEATURE] Gradient checkpointing in `forward_intermediates()`

4

**Is your feature request related to a problem? Please describe.** I rely on the `forward_intermediates()` API for object detection models, and I'm experimenting with ViT-g and would like to try...

collinmccarthy

Use HF Papers

1

qgallouedec

Hi, timm is excellent and helps me a lot for my research. Thanks for your great work! Do you have the plan to support more strong vision models like Depth...

Jacobew

[RFC] Faster load time for large models

24

While playing around with Flux-Redux, which used siglip-so400m, I noticed that loading time is pretty slow. I think we can bring some of the loading-time optimizations for LLMs to timm...

gau-nernst

Main train.py does not properly instantiate ImageFolder dataset if --train-split --val-split and --num-classes are specified. It looks like this may only be happening if the main data-dir had additional folders...

RorryB