cleaning up

This commit is contained in:
HimariO 2025-04-01 21:07:36 +08:00
parent c891300c1e
commit fdae70a832
3 changed files with 8 additions and 60 deletions

View file

@ -102,7 +102,7 @@ def main(args):
np_dtype = np.float32
ftype = 0
elif args.data_type == 'fp16':
dtype = torch.float32
dtype = torch.float16
np_dtype = np.float16
ftype = 1
else: