pytorch-image-models/timm
Ross Wightman 447677616f version 0.5.5 2022-01-20 21:18:30 -08:00
..
data Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones. 2022-01-14 16:28:27 -08:00
loss Update binary cross ent impl to use thresholding as an option (convert soft targets from mixup/cutmix to 0, 1) 2021-09-23 15:45:39 -07:00
models Last batch of small model weights (for now). mobilenetv3_small 050/075/100 and updated mnasnet_small with lambc/lamb optimizer. 2022-01-19 10:02:02 -08:00
optim fix lars 2022-01-19 17:49:43 +08:00
scheduler Fix k_decay default arg != 1.0 in poly scheduler 2021-11-22 09:37:36 -08:00
utils Add --fuser arg to train/validate/benchmark scripts to select jit fuser type 2022-01-17 13:54:25 -08:00
__init__.py Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models 2021-04-12 11:54:22 -07:00
version.py version 0.5.5 2022-01-20 21:18:30 -08:00