Skip to content

TextBrewer 0.1.8
Compare
Choose a tag to compare
@airaria airaria released this 11 Mar 01:07
· 127 commits to master since this release
v0.1.8
7e0bd0d

Improvements:

  • TrainingConfig.log_dir can be set to None to disable TensorBoard.
  • Added an attribute print_freq to the distiller to control the frequency of logging.
  • Added a new argument num_steps to the train method of the distiller. If num_steps is specified, the distiller will ignore num_epochs and allow an unknown-size dataloader (i.e., which has no __len__ attribute).
  • Added a new argument batch_postprocessor to the train method of the distiller to allow post-processing of batches.
Assets 3
Loading