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 |
|
limiteinductive
|
2ef4982e04
|
remove wandb from base config
|
2024-02-07 11:06:59 +01:00 |
|
limiteinductive
|
ea05f3d327
|
make device and dtype work in Trainer class
|
2024-02-06 23:10:10 +01:00 |
|
limiteinductive
|
0ee2d5e075
|
Fix warmup steps calculation when gradient_accumulation is used
|
2024-01-25 12:20:36 +01:00 |
|
limiteinductive
|
7f722029be
|
add basic unit test for training_utils
|
2024-01-14 22:08:20 +01:00 |
|