1dcbf4f64020281cd8a3e43193f44ee5cab3b777,examples/dqn_and_double_dqn.py,,experiment,#Any#,15

Before Change

        qf_criterion=qf_criterion,
        **variant["algo_params"]
    )
    if ptu.gpu_enabled():
        algorithm.cuda()
    algorithm.train()


if __name__ == "__main__":

After Change

        qf_criterion=qf_criterion,
        **variant["algo_params"]
    )
    algorithm.to(ptu.device)
    algorithm.train()


if __name__ == "__main__":

Instances