saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix training pipeline (#16342)

Browse source
This commit is contained in:
Baiju Meswani 2023-06-13 15:06:38 -07:00 committed by GitHub
parent c5176ed122
commit ed2482667b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
orttraining/orttraining/test/python/orttraining_test_ortmodule_bert_classifier.py
View file

@ -220,7 +220,7 @@ def load_dataset(args):
# Load the dataset into a pandas dataframe.
df = pd.read_csv(
os.path.join(args.data_dir, "in_domain_train.tsv"),
os.path.join(args.data_dir if os.path.exists(args.data_dir) else "cola_public/raw", "in_domain_train.tsv"),
delimiter="\t",
header=None,
names=["sentence_source", "label", "label_notes", "sentence"],

2
orttraining/orttraining/test/python/orttraining_test_ortmodule_bert_classifier_autocast.py
View file

@ -218,7 +218,7 @@ def load_dataset(args):
# Load the dataset into a pandas dataframe.
df = pd.read_csv(
os.path.join(args.data_dir, "in_domain_train.tsv"),
os.path.join(args.data_dir if os.path.exists(args.data_dir) else "cola_public/raw", "in_domain_train.tsv"),
delimiter="\t",
header=None,
names=["sentence_source", "label", "label_notes", "sentence"],