storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-18 07:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

update README; remove print; make style

Browse Source
This commit is contained in:
Aryan
2024-12-27 07:19:16 +01:00
parent 87af1a0113
commit a19401e84f
3 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/args.py
+3 -1
View File
@@ -60,7 +60,9 @@ class Args:
# Training arguments
training_type: str = None
seed: int = 42
mixed_precision: str = None # TODO: consider removing later https://github.com/a-r-r-o-w/finetrainers/pull/139#discussion_r1897438414
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