pretrained model prediction result

[BinghaoLu]

trafficstars

Hi, I just used the pretrained model you provided to evaluate on the sample dataset, the result doesn't seem to make reasonable prediction, I also tried to train the model by myself, but hasn't good prediction either, has anyone used the pretrained model to predict good result, or have you successfully train the model to make good result?

[Cirrusfloccus31]

Hi,

I think I have the same problem. Even if the input is a torch.zeros the model output has no sense ! Did someone has good results with the pretrained model ?

Thank you

[BinghaoLu]

I added sigmoid activation to the output during the training and evaluation and it produced reasonable results.