gucasbrg

move ``` mask = torch.prod(gt_params != 0, 1).byte() ``` to ``` mask = torch.prod(gt_params != 0, 1).bool() ```