PS: The paper trains the model for 100 epochs for a fair comparison. You can use more data and train for more epochs to get better performance.