limiteinductive
|
38c86f59f4
|
Switch gradient clipping to native torch torch.nn.utils.clip_grad_norm_
|
2024-03-19 22:08:48 +01:00 |
|
limiteinductive
|
432e32f94f
|
rename Scheduler -> LRScheduler
|
2024-02-15 11:48:36 +01:00 |
|
limiteinductive
|
cef8a9936c
|
refactor register_model decorator
|
2024-02-12 16:21:04 +01:00 |
|
limiteinductive
|
d6546c9026
|
add @register_model and @register_callback decorators
Refactor ClockTrainer to include Callback
|
2024-02-12 10:24:19 +01:00 |
|
limiteinductive
|
f541badcb3
|
Allow optional train ModelConfig + forbid extra input for configs
|
2024-02-10 16:13:10 +01:00 |
|
Pierre Colle
|
25bfa78907
|
lr, betas, eps, weight_decay at model level
Co-authored-by: Cédric Deltheil <355031+deltheil@users.noreply.github.com>
|
2024-02-09 12:05:13 +01:00 |
|