You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hello! I've found a performance issue in utils.py/: dataset.batch(self.batch_size,drop_remainder=batch_drop_remainder)(here) should be called before dataset.map(_parse_function)(here), which could make your program more efficient.
Besides, you need to check the function _parse_function called in dataset.map(_parse_function) whether to be affected or not to make the changed code work properly. For example, if _parse_function needs data with shape (x, y, z) as its input before fix, it would require data with shape (batch_size, x, y, z) after fix.
Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.
The text was updated successfully, but these errors were encountered:
Hello! I've found a performance issue in utils.py/:
dataset.batch(self.batch_size,drop_remainder=batch_drop_remainder)
(here) should be called beforedataset.map(_parse_function)
(here), which could make your program more efficient.Here is the tensorflow document to support it.
Besides, you need to check the function
_parse_function
called indataset.map(_parse_function)
whether to be affected or not to make the changed code work properly. For example, if_parse_function
needs data with shape (x, y, z) as its input before fix, it would require data with shape (batch_size, x, y, z) after fix.Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.
The text was updated successfully, but these errors were encountered: