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

2019-06-24 13:12:04 -07:00

..

data

Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py

2019-06-20 17:29:25 -07:00

loss

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

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

2019-06-24 13:12:04 -07:00

optim

Big re-org, working towards making pip/module as 'timm'

2019-06-19 17:20:51 -07:00

scheduler

Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py

2019-06-20 17:29:25 -07:00

__init__.py

Add model registry and model listing fns, refactor model_factory/create_model fn

2019-06-23 18:22:16 -07:00

utils.py

Add current checkpoints output back to CheckpointSaver (via logger)

2019-06-21 11:57:43 -07:00

version.py

Bump version to 0.1.2

2019-06-23 18:27:30 -07:00