refiners/tests/training_utils
2024-04-17 18:10:28 +02:00
..
mock_config.toml Switch gradient clipping to native torch torch.nn.utils.clip_grad_norm_ 2024-03-19 22:08:48 +01:00
mock_config_2_models.toml Switch gradient clipping to native torch torch.nn.utils.clip_grad_norm_ 2024-03-19 22:08:48 +01:00
test_common.py fix training_utils requirements check 2024-04-17 18:10:28 +02:00
test_metrics.py add DINOv2-FD metric 2024-04-03 16:45:00 +02:00
test_trainer.py TrainerClock: assert dataset_length >= batch_size 2024-04-12 15:05:52 +02:00