Update model.py

model.py

@@ -47,3 +47,8 @@ class GroupedAutoEncoder(nn.Module):
         # Concatenate groups back together
         # reconstructed = torch.cat(decoded_groups, dim=1)
         return reconstructed
+
+input_dim = 5120
+hidden_dim = 320
+num_groups = 40
+model = GroupedAutoEncoder(input_dim=input_dim, hidden_dim=hidden_dim, num_groups=num_groups).cuda()