1f720ee453871b2ab764f608926281716ef7bf81,tools/train_pl.py,,,#,340

Before Change

lit = LitModel(opt)
// warning grad_clip_mode is ignored.
trainer = pl.Trainer(
    callbacks=[OnEpochStartCallback()],
    default_root_dir=opt.checkpoint_path,
    resume_from_checkpoint=resume_from,
    distributed_backend="ddp",

After Change

        // progress_bar_refresh_rate=0,
        // fast_dev_run=True,
    )
    trainer.test(lit)
else:
    trainer.fit(lit)

Instances