Hi everyone! I am evaluating my network and I realize that my RMSE metric is not always the same. It depends of the batch_size.
Here is my code.:
data_loader = DataLoader(....., batch_size, True, workers, stage="test")
rmse = []
for batch_idx, sample in enumerate(data_loader):
sample['color'] = sample['color'].cuda()
sample['gt'] = sample['gt'].cuda()
mu = self.model(sample['color'])
sample['gt'] = sample['gt'].view(-1)
mu = mu.view(-1)
rmse.append( compute_rsme(sample['gt'], mu))
print(rmse.mean())
Any idea why? any help would be great =)