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

update (#5)

Browse Source
This commit is contained in:
Aryan
2024-10-06 21:53:43 +05:30
committed by GitHub
parent d81209bf49
commit d792cbbb64
5 changed files with 49 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/args.py
+5
View File
@@ -96,6 +96,11 @@ def _get_dataset_args(parser: argparse.ArgumentParser) -> None:
default=0,
help="Number of subprocesses to use for data loading. 0 means that the data will be loaded in the main process.",
)
parser.add_argument(
"--pin_memory",
action="store_true",
help="Whether or not to use the pinned memory setting in pytorch dataloader.",
)
def _get_validation_args(parser: argparse.ArgumentParser) -> None: