Skip to content

Commit 578fc59

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent d67b89f commit 578fc59

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

acceleration/dataset_type_performance.ipynb

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -511,7 +511,9 @@
511511
"set_determinism(seed=0)\n",
512512
"train_trans, val_trans = transformations()\n",
513513
"lmdb_init_start = time.time()\n",
514-
"train_lmdb_ds = LMDBDataset(data=train_files, transform=train_trans, cache_dir=LMDB_cache, lmdb_kwargs={\"map_async\": True})\n",
514+
"train_lmdb_ds = LMDBDataset(\n",
515+
" data=train_files, transform=train_trans, cache_dir=LMDB_cache, lmdb_kwargs={\"map_async\": True}\n",
516+
")\n",
515517
"val_lmdb_ds = LMDBDataset(data=val_files, transform=val_trans, cache_dir=LMDB_cache, lmdb_kwargs={\"map_async\": True})\n",
516518
"lmdb_init_time = time.time() - lmdb_init_start\n",
517519
"\n",
@@ -522,10 +524,7 @@
522524
" lmdb_metric_values,\n",
523525
" lmdb_epoch_times,\n",
524526
") = train_process(train_lmdb_ds, val_lmdb_ds)\n",
525-
"print(\n",
526-
" f\"total training time of {lmdb_epoch_num}\"\n",
527-
" f\" epochs with LMDB storage Dataset: {lmdb_total_time:.4f}\"\n",
528-
")"
527+
"print(f\"total training time of {lmdb_epoch_num}\" f\" epochs with LMDB storage Dataset: {lmdb_total_time:.4f}\")"
529528
]
530529
},
531530
{
@@ -549,8 +548,12 @@
549548
"set_determinism(seed=0)\n",
550549
"train_trans, val_trans = transformations()\n",
551550
"cache_init_start = time.time()\n",
552-
"cache_train_ds = CacheDataset(data=train_files, transform=train_trans, cache_rate=1.0, runtime_cache=\"processes\", copy_cache=False)\n",
553-
"cache_val_ds = CacheDataset(data=val_files, transform=val_trans, cache_rate=1.0, runtime_cache=\"processes\", copy_cache=False)\n",
551+
"cache_train_ds = CacheDataset(\n",
552+
" data=train_files, transform=train_trans, cache_rate=1.0, runtime_cache=\"processes\", copy_cache=False\n",
553+
")\n",
554+
"cache_val_ds = CacheDataset(\n",
555+
" data=val_files, transform=val_trans, cache_rate=1.0, runtime_cache=\"processes\", copy_cache=False\n",
556+
")\n",
554557
"cache_init_time = time.time() - cache_init_start\n",
555558
"\n",
556559
"(\n",

0 commit comments

Comments
 (0)