fix: comments

This commit is contained in:
NanoCode012
2025-02-21 14:00:45 +07:00
parent 1ea6ce73ed
commit a6ce7d7522

View File

@@ -214,7 +214,7 @@ class TestHFRLTrainerBuilder:
assert training_arguments.warmup_steps == 10
assert training_arguments.cosine_min_lr_ratio == 0.1
# Hardware settings
# Other settings
assert training_arguments.dataloader_num_workers == 1
assert training_arguments.dataloader_pin_memory is True
assert training_arguments.gradient_checkpointing is False