We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d4c29dd commit 7f42009Copy full SHA for 7f42009
tools/paddle_infer.py
@@ -63,11 +63,13 @@ def init_predictor(args):
63
64
def create_data_loader(args):
65
data_dir = args.data_dir
66
- reader_file = args.reader_file.split(".")[0]
+ reader_path, reader_file = os.path.split(args.reader_file)
67
+ reader_file, extension = os.path.splitext(reader_file)
68
batchsize = args.batchsize
69
place = args.place
70
file_list = [os.path.join(data_dir, x) for x in os.listdir(data_dir)]
- sys.path.append(os.path.abspath("."))
71
+ sys.path.append(reader_path)
72
+ #sys.path.append(os.path.abspath("."))
73
reader_class = import_module(reader_file)
74
dataset = reader_class.RecDataset(file_list, config=None)
75
loader = DataLoader(
0 commit comments