'dual_encoder_ranking' object has no attribute 'final_response_output'

jasonwu0731 / ToD-BERT

Pre-Trained Models for ToD-BERT

BSD 2-Clause "Simplified" License

291 stars 55 forks source link

Closed priyanksonis closed 4 years ago

priyanksonis commented 4 years ago

While evaluation on response selection with batch_size < 100, on this line: https://github.com/jasonwu0731/ToD-BERT/blob/master/models/dual_encoder_ranking.py#L95 we will get the error: 'dual_encoder_ranking' object has no attribute 'final_response_output' , so please check this.

jasonwu0731 commented 4 years ago

Updated it with args["eval_batch_size"] as shown here. Thanks.