loss fix (#25)

Author: cohimame

saicinpainting/evaluation/losses/base_loss.py

@@ -64,7 +64,7 @@ def get_value(self, groups=None, states=None):
             group_results: None, if groups is None;
                 else dict {group_idx: {'mean': score mean among group, 'std': score std among group}}
         """
-        individual_values = torch.stack(states, dim=0).reshape(-1).cpu().numpy() if states is not None \
+        individual_values = torch.cat(states, dim=-1).reshape(-1).cpu().numpy() if states is not None \
             else self.individual_values
 
         total_results = {