feat: 'README'

modeling_flame.py

@@ -709,7 +709,7 @@ class FLAMEModel(nn.Module):
         # calculate loss
         point_loss = self.point_loss(forecasts, target)
         rec_loss = self.point_loss(x_rec, input)
-        transformed_target = rearrange(target, 'b (n p) c -> (b c n) p', p=self.patch_len)
+        transformed_target = rearrange(target, 'b (n p) -> (b n) p', p=self.patch_len)
         raw_prob_loss = -dist.log_prob(transformed_target)
         mask = raw_prob_loss < eps
         raw_prob_loss = torch.where(mask, raw_prob_loss, torch.zeros_like(raw_prob_loss))