Require gradient false in eval mode

henryhungle / MTN

Code for the paper Multimodal Transformer Networks for End-to-End Video-Grounded Dialogue Systems (ACL19)

MIT License

98 stars 25 forks source link

Require gradient false in eval mode #7

Closed hwanheelee1993 closed 3 years ago

hwanheelee1993 commented 4 years ago

In your code, you did not wrap the eval part with "torch.no_grad()", which means that your code computes gradient for validation data. That leads to unnecessary computation... You'd better edit that part.

henryhungle commented 3 years ago

Thanks! I updated the code and will close this