ResidualAttentionNetwork-pytorch
ResidualAttentionNetwork-pytorch copied to clipboard
model = ResidualAttentionModel() error with python3
TypeError: new() received an invalid combination of arguments - got (float, int, int, int), but expected one of:
- (torch.device device)
- (torch.Storage storage)
- (Tensor other)
- (tuple of ints size, torch.device device)
- (object data, torch.device device)
You need to convert the input and output filters parameters of the Conv2d layers to int. If you look at the current arguments, they are float ( eg : 2.0 instead of 2) which is not accepted by torch.