Ruiyang Ren

Hi all, thanks for the cool contribution. I noticed that `TPUEstimator` is used in the code. After running on a machine with multiple GPUs, I found that only one GPU...

Training data for NQ?

6

Thanks for the great contribution! I found that the downloaded data of NQ only contains test files and corpus, where can I get the training files? Thank you!

TypeError when running Quick Example

1

Thanks for the great work. When I run the Quick Example, I got a TypeError: ``` Traceback (most recent call last): File "demo.py", line 34, in ndcg, _map, recall, precision...

您好，感谢您的分享。 请问如果我想在代码中加入pytorch的重计算机制（torch.utils.checkpoint)来减少显存开销，应该怎么做呢？