d2l-ai / d2l-vi

Một cuốn sách về Học Sâu đề cập đến nhiều framework phổ biến, được sử dụng trên 300 trường Đại học từ 55 đất nước bao gồm MIT, Stanford, Harvard, và Cambridge.
Other
615 stars 245 forks source link

Performance issues in d2l/tensorflow.py(P2) #3127

Closed DLPerf closed 2 years ago

DLPerf commented 3 years ago

Hello,I found a performance issue in the definition of load_data_fashion_mnist , d2l/tensorflow.py, .map(resize_fn)) was called without num_parallel_calls. I think it will increase the efficiency of your program if you add this.

Here is the documemtation of tensorflow to support this thing.

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