Replace hard coded train, dev and test file names with args
1 file changed