Commit Graph

6 Commits (506df0e3d0136da4063b2a2881958e57cf43c784)

Author SHA1 Message Date
Ross Wightman 3d9c8a6489 Add support for new AMP checkpointing support w/ amp.state_dict
5 years ago
Ross Wightman 6cdf35e670 Add explicit half/fp16 support to loader and validation script
6 years ago
Ross Wightman c6b32cbe73 A number of tweaks to arguments, epoch handling, config
6 years ago
Ross Wightman 827a3d6010 Add current checkpoints output back to CheckpointSaver (via logger)
6 years ago
Ross Wightman 6fc886acaf Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py
6 years ago
Ross Wightman aa4354f466 Big re-org, working towards making pip/module as 'timm'
6 years ago