Skip to content

Torch_Notes.md

Latest commit

 

History

History
32 lines (19 loc) · 688 Bytes

Torch_Notes.md

File metadata and controls

32 lines (19 loc) · 688 Bytes
Raw

Pytorch

tags # torch

  • set device

tags # torch device # torch device if cuda # torch get device # torch device variable

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

lightning

  • progress bar refresh, reduce freeze

tags # progress bar refresh rate # progress bar freeze # torch lightning progress refresh # progress bar stuck

  1. set refresh rate before constructing progress_bar

  2. put into callbacks

mnist_model = MNISTModel()

# prepare progress bar, set refresh rate
our_progress_bar = TQDMProgressBar(refresh_rate=10)
trainer = pl.Trainer(max_epochs=10, callbacks=[our_progress_bar])

trainer.fit(mnist_model)