Shen Guangyao
Results
2
comments of
Shen Guangyao
I think it should be: ``` outputs, targets = [], [] with torch.no_grad(): for batch_idx, (data, target) in enumerate(self.valid_data_loader): data, target = data.to(self.device), target.to(self.device) output = self.model(data) pred = torch.argmax(output,...
> @sunlightsgy , I get your point. Assume that we have a dataset size N, each image has size M x L, the batch_size is B. If N%B = 0,...