-
Notifications
You must be signed in to change notification settings - Fork 113
Open
Description
Hello! I've found a performance issue in your project: 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:
- /7. ResNet/train.py:
.batch(128)(here) should be called before.map(preprocess)(here). - /7. ResNet/train.py:
.batch(128)(here) should be called before.map(preprocess)(here). - /4. AlexNet/AlexNet.py:
.batch(32)(here) should be called before.map(preprocess)(here). - /5. VGG11/VGG11.py:
.batch(16)(here) should be called before.map(preprocess)(here).
Besides, you need to check the function called in map()(e.g., preprocess called in .map(preprocess)) whether to be affected or not to make the changed code work properly. For example, if preprocess 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.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels