search

ahmetgunduz / Real-time-GesRec

Real-time Hand Gesture Recognition with PyTorch on EgoGesture, NvGesture, Jester, Kinetics and UCF101
https://arxiv.org/abs/1901.10323
MIT License
619 stars 168 forks source link

pushed model inside no_grad., added .item() for previous_video_id #88

Closed badri2304 closed 3 years ago

badri2304 commented 3 years ago

model(inputs) have to be pushed inside no_grad else it occupies more gpu memory. And previous_video_id.item() has to be there, else the keys for results world be tensor(0), tensor(1).... which gives error when doing json.dump()

ahmetgunduz commented 3 years ago

thanks for update.