About the metric on Tracffic BJ?

[toddwyl]

The shape of data in dataset Traffic BJ is 2 channels. (shape: [B, 2, 32, 32]) The mse in Traffic BJ are the mean of channels or the sum of the channels ? Because I noticed that the code in repo is written as below:

np.mean((predictions-target)**2 , axis=(0,1,2)).sum()