This commit is contained in:
sayakpaul
2024-12-26 08:01:16 +05:30
parent 0309f3bf67
commit 1d5e0566fb
2 changed files with 23 additions and 23 deletions
+1 -1
View File
@@ -60,7 +60,7 @@ class Args:
# Training arguments
training_type: str = None
seed: int = 42
mixed_precision: str = None
mixed_precision: str = None # TODO: consider removing later https://github.com/a-r-r-o-w/finetrainers/pull/139#discussion_r1897438414
batch_size: int = 1
train_epochs: int = 1
train_steps: int = None