miniworld.utils.DataClass.MiniWorldTemplateClass¶

class miniworld.utils.DataClass.MiniWorldTemplateClass(template_dict=None)[source]¶

__init__(template_dict=None)[source]¶

template_dict :

{: ‘xyz’ : torch.tensor (params[‘N_PICK_GLOBAL’], L_crop, 27, 3) ‘template_1D’ : torch.tensor (params[‘N_PICK_GLOBAL’], L_crop, NUM_CLASSES + 1) ‘template_atom_mask’: torch.tensor (params[‘N_PICK_GLOBAL’], L_crop, 27)

}

Methods

__init__([template_dict])

template_dict :