chaton
|
c9622bafe0
|
[feat] Add Trainer(stochastic_weight_avg=True/False) (#6038)
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
|
2021-02-17 23:10:05 +00:00 |
|
Jirka Borovec
|
9475c845cb
|
Docs/fixes (#5914)
* wip
* ..
* ...
* Apply suggestions from code review
Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com>
Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com>
|
2021-02-11 10:22:07 +00:00 |
|
chaton
|
7b00894130
|
[feat] Add StochasticWeightAveragingCallback (#5640)
* add swa callback
* switch back to 1.6.0
* remove optimizer_step
* move super
* update
* forgot update_parameters
* update on comments
* works for ddp
* resolve flake8
* remove set_model
* resolve flake8
* resolve cpu
* resolve flake8
* resolve flake8
* update
* update on comments
|
2021-02-11 00:05:59 +00:00 |