Function print_shape

easyvolcap/utils/net_utils.py:30–41

(batch: dotdict)

def print_shape(batch: dotdict):
    if isinstance(batch, dict):
        for k, v in batch.items():
            print(k)
            print_shape(v)
    elif isinstance(batch, list):
        for v in batch:
            print_shape(v)
    elif isinstance(batch, torch.Tensor):
        print(f'{batch.shape}')
    else:
        print(batch)


type_mapping = {