You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/timm
Ross Wightman 13c19e213d
Add native PyTorch EfficientNet B1 and B2 weights. Not quite where I want them, but hitting the brick wall and moving on to other projects...
5 years ago
..
data Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py 5 years ago
loss Big re-org, working towards making pip/module as 'timm' 5 years ago
models Add native PyTorch EfficientNet B1 and B2 weights. Not quite where I want them, but hitting the brick wall and moving on to other projects... 5 years ago
optim Big re-org, working towards making pip/module as 'timm' 5 years ago
scheduler Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py 5 years ago
__init__.py Add model registry and model listing fns, refactor model_factory/create_model fn 5 years ago
utils.py Add current checkpoints output back to CheckpointSaver (via logger) 5 years ago
version.py Bump version to 0.1.2 5 years ago