Fix a bug of replay

2026-09-09 11:13:47 +08:00 · 2017-05-26 15:18:45 -06:00
parent 45a3f00d32
commit d77cc46cbf
4 changed files with 17 additions and 42 deletions
@@ -26,6 +26,7 @@ class FullyConnectedNet(nn.Module):
            print 'Network transferred.'

    def forward(self, x):
+        x = x.reshape((x.shape[0], -1))
        x = torch.from_numpy(np.asarray(x, dtype='float32'))
        if self.gpu:
            x = x.cuda()