refiners/tests/training_utils
2024-04-13 15:03:35 +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 turn scoped_seed into a context manager 2024-04-13 15:03:35 +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