* fix get_num_masks output as [int] to int * fix loss size from torch.Size([1]) to torch.Size([])
Styling