Open chengchingwen opened 6 years ago
https://github.com/YichenGong/Densely-Interactive-Inference-Network/blob/b36b9220dc16084240bebed51c575d43a8529f12/python/train_mnli.py#L478 save_submission(path, IDs, logits[1:]) should be just save_submission(path, IDs, logits)
save_submission(path, IDs, logits[1:])
save_submission(path, IDs, logits)
As
https://github.com/YichenGong/Densely-Interactive-Inference-Network/blob/b36b9220dc16084240bebed51c575d43a8529f12/python/train_mnli.py#L477
have sliced logits already
https://github.com/YichenGong/Densely-Interactive-Inference-Network/blob/b36b9220dc16084240bebed51c575d43a8529f12/python/train_mnli.py#L478
save_submission(path, IDs, logits[1:])should be justsave_submission(path, IDs, logits)As
https://github.com/YichenGong/Densely-Interactive-Inference-Network/blob/b36b9220dc16084240bebed51c575d43a8529f12/python/train_mnli.py#L477
have sliced logits already