vietdoan / enet_pytorch Goto Github PK

hi, @vietdoan when i trainiing the enet encoder model use 2 gpu?
but i got the error:RuntimeError: tensors are on different GPUs!
i try to solve this problem, but failed?could help me
thank you very much

Hi @vietdoan , thank you for sharing your implementation of ENet. Currently, I am trying to train the model with CamVid dataset. The class accuracy averages to 0.501, while the class iou averages to 0.321. But the result I obtained is far worse than the authors'. I am not sure if it is possible to share your model parameters? Thank you.

Hello ,could you help me how to calculate the segmentation speed.I don't know how to calculate the speed in pytorch. Thank you!

could you provide the pretrained model (enet_encoder_camvid_299.pkl)?

thank you