BsiNet-torch

BsiNet-torch copied to clipboard

Tensor size mismatch during prediction

1

I received tensor mismatch error when making a prediction. `inputs = inputs.to(device = device)`\ `outputs1, outputs2, outputs3 = model(inputs)` 

kcsumesh

pretrained weight

3

Could you please upload the pretrained weight to google drive Thank you

mos3ab8