Ross Wightman
|
e685618f45
|
Merge pull request #550 from amaarora/wandb
Wandb Support
|
2021-04-15 09:26:35 -07:00 |
Ross Wightman
|
7c97e66f7c
|
Remove commented code, add more consistent seed fn
|
2021-04-12 09:51:36 -07:00 |
Aman Arora
|
5772c55c57
|
Make wandb optional
|
2021-04-10 01:34:20 -04:00 |
|
Aman Arora
|
f54897cc0b
|
make wandb not required but rather optional as huggingface_hub
|
2021-04-10 01:27:23 -04:00 |
|
Aman Arora
|
3f028ebc0f
|
import wandb in summary.py
|
2021-04-08 03:48:51 -04:00 |
|
Aman Arora
|
624c9b6949
|
log to wandb only if using using wandb
|
2021-04-08 03:40:22 -04:00 |
|
Aman Arora
|
6b18061773
|
Add GIST to docstring for quick access
|
2021-03-29 15:33:31 +11:00 |
|
Aman Arora
|
92b1db9a79
|
update docstrings and add check on and
|
2021-03-29 10:04:51 +11:00 |
|
Aman Arora
|
b85be24054
|
update to work with fnmatch
|
2021-03-29 09:36:31 +11:00 |
|
Aman Arora
|
20626e8387
|
Add to extract stats for SPP
|
2021-03-27 05:40:04 +11:00 |
|
Ross Wightman
|
4f49b94311
|
Initial AGC impl. Still testing.
|
2021-02-15 23:22:44 -08:00 |
|
Ross Wightman
|
4203efa36d
|
Fix #387 so that checkpoint saver works with max history of 1. Add checkpoint-hist arg to train.py.
|
2021-01-31 20:14:51 -08:00 |
|
Ross Wightman
|
4ca52d73d8
|
Add separate set and update method to ModelEmaV2
|
2020-12-03 10:05:09 -08:00 |
|
Ross Wightman
|
27bbc70d71
|
Add back old ModelEma and rename new one to ModelEmaV2 to avoid compat breaks in dependant code. Shuffle train script, add a few comments, remove DataParallel support, support experimental torchscript training.
|
2020-11-29 16:22:19 -08:00 |
|
Ross Wightman
|
9214ca0716
|
Simplifying EMA...
|
2020-11-16 12:51:52 -08:00 |
|
Ross Wightman
|
4a3df7842a
|
Fix topn metric view regression on PyTorch 1.7
|
2020-10-29 14:04:15 -07:00 |
|
Ross Wightman
|
80078c47bb
|
Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.
|
2020-10-09 17:24:43 -07:00 |
|
Ross Wightman
|
fcb6258877
|
Add missing leaky_relu layer factory defn, update Apex/Native loss scaler interfaces to support unscaled grad clipping. Bump ver to 0.2.2 for pending release.
|
2020-10-02 16:19:39 -07:00 |
|
Ross Wightman
|
532e3b417d
|
Reorg of utils into separate modules
|
2020-09-07 13:58:09 -07:00 |