pangpd

icon location Just a boy caught up in dreams and fantasies

Results 1 comments of pangpd

is cuda() type data are not added to the isinstance list?

You only need to provide shape, that is, "temp = (3, 224, 224)", not "temp= torch.tensor (temp).to(device)"