minor fix to ensure squeeze works
@@ -240,7 +240,7 @@ class StarspaceAgent(Agent):
def same(self, y1, y2):
if len(y1.squeeze(0)) != len(y2.squeeze(0)):
return False
- if abs((y1.squeeze()-y2.squeeze()).sum().data.sum()) > 0.00001:
+ if abs((y1.squeeze(0)-y2.squeeze(0)).sum().data.sum()) > 0.00001:
return False
return True
