OneCycle LR set in Pytorch lightning

 

Add configure_optimizer member function in pytorch lightning model class.

refer to code:

..

def configure_optimizers(self):

optimizer = getattr(torch.optim, self.cfg.optimizer)

self.optimizer = optimizer(self.parameters(), lr=float(self.cfg.lr))

total_bs = int(self.cfg.train_dataloader_len / self.cfg.gpus)

epochs = self.cfg.epochs

self.scheduler = torch.optim.lr_scheduler.OneCycleLR(

self.optimizer, max_lr=self.cfg.lr,

anneal_strategy='linear', div_factor=100,

steps_per_epoch=total_bs, pct_start=(1/self.cfg.epochs),

epochs=epochs)

sched = {

'scheduler': self.scheduler,

'interval': 'step',

}

return [self.optimizer], [sched]

..

Thank you. 🙇🏻‍♂️

www.marearts.com

python list reverse

refer to code: 

..

Test_list = [1, 2, 3, 4, 5]
Test_list.reverse()
print(Test_list)  # Output:  [5,4,3,2,1]

..

Thank you.

re-install nvidia drive (cuda) in ubuntu

command it on terminal

 

..

sudo apt clean
sudo apt update
sudo apt purge nvidia-* 
sudo apt autoremove
sudo apt install -y cuda

..

Thank you.

www.marearts.com