Skip to content

Commit cd3de1d

Browse files
Test 02 branch (#906)
* yifei add dnn register part for uapi * yifei add register dnn part * yifei add rec runner.py * [add function] yifei add rec runner.py * yifei delete print funcs * yifei change gpu list length from 8 to 2 * yifei add gpu list * yifei change device name * yifei change dataset part * yifei change data check funcs --------- Co-authored-by: wangzhen38 <41941775+wangzhen38@users.noreply.github.com>
1 parent ef86749 commit cd3de1d

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

uapi_rec/rank/check_dataset.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,9 @@ def check_dataset(model_name, dataset_dir, dataset_type):
6565
# Use space as delimiter
6666
# TODO: Support more delimiters
6767
delim = ' '
68-
parts = line.rstrip().split(delim)
68+
parts = line.rstrip().split(delim)[0]
6969
label = parts.split(":")[1]
70+
label = int(label)
7071
if label not in (0, 1):
7172
raise CheckFailedError(
7273
f"The label of data in each row should be 0 or 1, but received {parts[0]}."
@@ -85,6 +86,8 @@ def check_dataset(model_name, dataset_dir, dataset_type):
8586
stagelog.success_datacheck(
8687
stage_id,
8788
train_dataset=meta['train.samples'],
88-
validation_dataset=meta['val.samples'],
89-
test_dataset=meta['test.samples'] or 0)
89+
validation_dataset=0,
90+
test_dataset=0)
91+
#validation_dataset=meta['val.samples'],
92+
#test_dataset=meta['test.samples'] or 0)
9093
return meta

0 commit comments

Comments
 (0)