henryhungle / MTN

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Require gradient false in eval mode

hwanheelee1993 opened this issue · comments

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.

Thanks! I updated the code and will close this