search

SaynaEbrahimi / UCB

Original PyTorch implementation of Uncertainty-guided Continual Learning with Bayesian Neural Networks, ICLR 2020
https://openreview.net/pdf?id=HklUCCVKDB
MIT License
73 stars 11 forks source link

Update ucb.py #7

Open mahi97 opened 3 years ago

mahi97 commented 3 years ago

Change .cuda to .to(device) to make it compatible with GPUs other than cuda:0