UCB
UCB copied to clipboard
Update ucb.py
Change .cuda to .to(device) to make it compatible with GPUs other than cuda:0