Kyumin Hwang
Kyumin Hwang
i also suffered from Loss:Nan.. I reduce learning rate from 1e-3 to 1e-1, but results is same(Loss : nan). If Loss is nan, then cannot store weights. so model cant...
@wave-transmitter Thank you for comment ! i solved this problem using learning rate. i reduced learning rate to 1e-5, then it worked correctly !
@ilovekj i recommend to find your proper learning rate ! i control to several times, and found proper rate. how about augment your dataset ?
@ilovekj i also used this code and i got efficient performance. In this code has augmentation module so that this code should make dataset more useful. how about increase to...
@gyfastas Thank you for comment. ids = score > CLS_THRESH if ids.numel()
@gyfastas Thank you for comment ! I also handle this error like return torch.tensor([[0.,0.,0.,0.]]), torch.tensor([[0]]) . Thank you :)