Update data_loader.py

[baixianger]

collate_fn函数内分离src和tgt的方式可能有误。