Chen Wu
Results
2
issues of
Chen Wu
This PR adds an argument `input_keys` for semantic search. If provided, the search is based on the input variables instead of all variables. To be consistent with the previous versions,...
Problem description: the current preprocessing function will squeeze length-1 sequences to a 0-dim tensor, which will cause an error in pytorch dataloaders. Solution: replaced each squeeze() with squeeze(0) in seq2seq_utils.py