Fix errors with irregular batch sizes during inferences

mingkaid / rl-prompt

Accompanying repo for the RLPrompt paper

MIT License

286 stars 52 forks source link

Fix errors with irregular batch sizes during inferences #16

Closed mingkaid closed 1 year ago

mingkaid commented 1 year ago

Fixed the bug raised in Issue #15, where during training for text style transfer on the Shakespeare dataset, an assertion statement would be triggered due to irregular batch sizes