l11x0m7 / Question_Answering_Models

This repo collects and re-produces models related to domains of question answering and machine reading comprehension
MIT License
155 stars 45 forks source link

Performance issues in /MRC (by P3) #5

Open DLPerf opened 3 years ago

DLPerf commented 3 years ago

Hello! I've found a performance issue in /MRC: batch() should be called before map(), which could make your program more efficient. Here is the tensorflow document to support it.

Detailed description is listed below:

Besides, you need to check the function called in map()(e.g., parser called in .map(parser, num_parallel_calls=num_threads)) whether to be affected or not to make the changed code work properly. For example, if parser needs data with shape (x, y, z) as its input before fix, it would require data with shape (batch_size, x, y, z).

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.

DLPerf commented 2 years ago

Hello, I'm looking forward to your reply~