refiners/src/refiners/training_utils
2024-03-19 22:08:48 +01:00
..
__init__.py Switch gradient clipping to native torch torch.nn.utils.clip_grad_norm_ 2024-03-19 22:08:48 +01:00
callback.py ruff 3 formatting (Rye 0.28) 2024-03-08 10:42:05 +01:00
clock.py change TimeValue to a dataclass 2024-03-19 14:49:24 +01:00
common.py Switch gradient clipping to native torch torch.nn.utils.clip_grad_norm_ 2024-03-19 22:08:48 +01:00
config.py Switch gradient clipping to native torch torch.nn.utils.clip_grad_norm_ 2024-03-19 22:08:48 +01:00
huggingface_datasets.py ruff 3 formatting (Rye 0.28) 2024-03-08 10:42:05 +01:00
trainer.py Switch gradient clipping to native torch torch.nn.utils.clip_grad_norm_ 2024-03-19 22:08:48 +01:00
wandb.py Switch gradient clipping to native torch torch.nn.utils.clip_grad_norm_ 2024-03-19 22:08:48 +01:00