Hyperparameters: 1 epoch, max_len_dict, and 1e-5 learning rate