Fixed prefetching for the Triplet training

parent 4c0c1b1e
......@@ -38,7 +38,7 @@ class Triplet(Base):
return [self.data_placeholder, self.data2_placeholder, self.data3_placeholder]
def get_placeholders_triplet_forprefetch(self, name=""):
def get_placeholders_forprefetch(self, name=""):
"""
Returns a place holder with the size of your batch
"""
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment