This code is modified from two repositories:
Rainbow: https://github.com/Kaixhin/Rainbow
AdaBelief: https://github.com/juntang-zhuang/Adabelief-Optimizer
adabelief-pytorch==0.0.5
sh run.sh
See documentation in https://github.com/Kaixhin/Rainbow
Results are in the folder results/default
. We provide the results for two independent runs with AdaBelief.