target_diff2 = target_diff // 2
if target_diff2 < 0:
print("Model over-pools to %d for target_length %d." % \
(current_length, self.target_length), file=sys.stderr)
exit(1)
elif target_diff2 > 0:
current = layers.SliceCenter(
After Change
// heads
//////////////////////////////////////////////////////////////////////////////////////////////////////
head_keys = natsorted([v for v in vars(self) if v.startswith("head")])
self.heads = [getattr(self, hk) for hk in head_keys]
self.head_output = []
for hi, head in enumerate(self.heads):
if not isinstance(head, list):
