Pytorch-lightning: How to disable progress bar and printing model structure?

Created on 1 May 2020  ยท  2Comments  ยท  Source: PyTorchLightning/pytorch-lightning

โ“ Questions and Help

Before asking:

  1. search the issues.
  2. search the docs.

What is your question?

Code

What have you tried?

What's your environment?

  • OS: [e.g. iOS, Linux, Win]
  • Packaging [e.g. pip, conda]
  • Version [e.g. 0.5.2.1]
question

Most helpful comment

Progress bar can be disabled by setting the progress_bar_refresh_rate argument in trainer:

# don't show progress bar
trainer = Trainer(progress_bar_refresh_rate=0)

To disable model structure printing you can set the weights_summary argument in trainer:

# don't print a summary
trainer = Trainer(weights_summary=None)

please see the documentation on trainer flags for more info.

All 2 comments

Progress bar can be disabled by setting the progress_bar_refresh_rate argument in trainer:

# don't show progress bar
trainer = Trainer(progress_bar_refresh_rate=0)

To disable model structure printing you can set the weights_summary argument in trainer:

# don't print a summary
trainer = Trainer(weights_summary=None)

please see the documentation on trainer flags for more info.

thanks @SkafteNicki

Was this page helpful?
0 / 5 - 0 ratings

Related issues

jcreinhold picture jcreinhold  ยท  3Comments

justusschock picture justusschock  ยท  3Comments

srush picture srush  ยท  3Comments

williamFalcon picture williamFalcon  ยท  3Comments

DavidRuhe picture DavidRuhe  ยท  3Comments