hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

nit

Browse Source
This commit is contained in:
Remi Cadene
2024-12-03 17:24:55 +01:00
parent afbd42d082
commit c6e9a3dc24
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/manage_dataset.py
View File

@@ -57,7 +57,7 @@ if __name__ == "__main__":
parser_conso.add_argument( parser_conso.add_argument(
"--batch-size", "--batch-size",
type=int, type=int,
default=32, default=8,
help="Batch size loaded by DataLoader for computing the dataset statistics.", help="Batch size loaded by DataLoader for computing the dataset statistics.",
) )
parser_conso.add_argument( parser_conso.add_argument(
@@ -85,7 +85,7 @@ if __name__ == "__main__":
"--private", "--private",
type=int, type=int,
default=0, default=0,
help="Create a private dataset repository on the Hugging Face Hub.", help="Create a private dataset repository on the Hugging Face Hub. Push publicly by default.",
) )
args = parser.parse_args() args = parser.parse_args()