Skip to content

Commit 0f838b3

Browse files
authored
Add support for IGBH calibration dataset (#187)
* add support for calibration dataset * fix typo
1 parent 08ce312 commit 0f838b3

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

script/get-dataset-igbh/customize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def preprocess(i):
4343
run_cmd += x_sep + \
4444
f"""{
4545
env['MLC_PYTHON_BIN_WITH_PATH']} tools/split_seeds.py --path {download_loc} --dataset_size {
46-
env['MLC_DATASET_IGBH_SIZE']} """
46+
env['MLC_DATASET_IGBH_SIZE']} {env.get('MLC_IGBH_CALIBRATION_FLAG', '')} """
4747

4848
# compress graph(for glt implementation)
4949
if env.get('MLC_IGBH_GRAPH_COMPRESS', '') == "yes":

script/get-dataset-igbh/meta.yaml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -437,6 +437,9 @@ variations:
437437
glt:
438438
env:
439439
MLC_IGBH_GRAPH_COMPRESS: yes
440+
calibration:
441+
env:
442+
MLC_IGBH_CALIBRATION_FLAG: --calibration
440443
csc:
441444
group: compressed-layout
442445
default: true

0 commit comments

Comments
 (0)