ae003338600bb9188fa5a1901a3a3ecc8fda351c,ptan/agent.py,,default_states_preprocessor,#,38

Before Change

        np_states = np.expand_dims(states[0], 0)
    else:
        np_states = np.array([np.array(s, copy=False) for s in states], copy=False)
    v = Variable(torch.from_numpy(np_states), volatile=volatile)
    if cuda:
        v = v.cuda(device=device_id)
    return v


def float32_preprocessor(states, cuda=False, device_id=None, volatile=False):
    np_states = np.array(states, dtype=np.float32)

After Change

        np_states = np.expand_dims(states[0], 0)
    else:
        np_states = np.array([np.array(s, copy=False) for s in states], copy=False)
    return torch.tensor(np_states)


def float32_preprocessor(states):
    np_states = np.array(states, dtype=np.float32)

Instances