656933c471e64ad697be749b98cea93a758ac5cb,ch16/02_cheetah_es.py,,worker_func,#,128

Before Change

def worker_func(worker_id, params_queue, rewards_queue, cuda, noise_std):
    env = make_env()
    net = Net(env.observation_space.shape[0], env.action_space.shape[0])
    net.eval()
    if cuda:
        net.cuda()

After Change

def worker_func(worker_id, params_queue, rewards_queue, device, noise_std):
    env = make_env()
    net = Net(env.observation_space.shape[0], env.action_space.shape[0]).to(device)
    net.eval()

    while True:
        params = params_queue.get()

Instances