12 Commits (f7325c7b712100f79a9ab4ae54118d259c11bacf)

Author	SHA1	Message	Date
Ross Wightman	f7325c7b71	Support either deepspeed or fvcore for flop profiling	3 years ago
Ross Wightman	66253790d4	Add `--bench profile` mode for benchmark.py to just run deepspeed detailed profile on model	3 years ago
Ross Wightman	13a8bf7972	Add train size override and deepspeed GMACs counter (if deepspeed installed) to benchmark.py	3 years ago
Ross Wightman	ac469b50da	Optimizer improvements, additions, cleanup ... * Add MADGRAD code * Fix Lamb (non-fused variant) to work w/ PyTorch XLA * Tweak optimizer factory args (lr/learning_rate and opt/optimizer_name), may break compat * Use newer fn signatures for all add,addcdiv, addcmul in optimizers * Use upcoming PyTorch native Nadam if it's available * Cleanup lookahead opt * Add optimizer tests * Remove novograd.py impl as it was messy, keep nvnovograd * Make AdamP/SGDP work in channels_last layout * Add rectified adablief mode (radabelief) * Support a few more PyTorch optim, adamax, adagrad	3 years ago
Ross Wightman	137a374930	Merge pull request #555 from MichaelMonashev/patch-1 ... benchmark.py argument description fixed	4 years ago
Ross Wightman	e15e68d881	Fix #566, summary.csv writing to pwd on local_rank != 0. Tweak benchmark mem handling to see if it reduces likelihood of 'bad' exceptions on OOM.	4 years ago
Michael Monashev	0be1fa4793	Argument description fixed	4 years ago
Ross Wightman	37c71a5609	Some further create_optimizer_v2 tweaks, remove some redudnant code, add back safe model str. Benchmark step times per batch.	4 years ago
Ross Wightman	288682796f	Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7	4 years ago
Ross Wightman	4445eaa470	Add img_size to benchmark output	4 years ago
Ross Wightman	0706d05d52	Benchmark models listed in txt file. Add more hybrid vit variants for testing	4 years ago
Ross Wightman	0e16d4e9fb	Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface.	4 years ago