Develop(8)
-
PyTorch Autograd Profiler
링크: https://pytorch.org/docs/stable/autograd.html#torch.autograd.profiler.profile Automatic differentiation package - torch.autograd — PyTorch master documentation Automatic differentiation package - torch.autograd torch.autograd provides classes and functions implementing automatic differentiation of arbitrary scalar valued functions. It requires minimal changes to the existing code - you only ..
2020.02.02 -
[Pytorch Error] RuntimeError: Given input size: (256x1x1). Calculated output size: (256x0x0). Output size is too small
Issue RuntimeError: Given input size: (256x1x1). Calculated output size: (256x0x0). Output size is too small input tensor의 크기가 작아서 생기는 문제이다. input tensor의 shape을 조정해주면 해결 가능하다. Solution transform = transforms.Compose([ transforms.Resize(256), transforms.RandomHorizontalFlip(), transforms.ToTensor(), transforms.Normalize( mean=model.original_model_info.mean, std=model.original_model_info.std ), ]..
2020.01.21