Is it correct that the obj_label in the demo is set to zero? Also, how can I extract the intermediate features?
'''
obj_label = meta_data['object_label'][cfg.rlbench.frame_id]
if obj_label == 0:
model_inputs.update({
'object_inputs': torch.rand(1024, 6),
'ee_pose': torch.eye(4),
'bottom_center': torch.zeros(3),
'object_center': torch.zeros(3),
'task': 'pick'
})
'''