①取消训练时的半精度
注释 torch_utils.py中的
x['model'].half() # to FP16
②取消验证时的半精度
修改engine/validator.py中的105行为
self.args.half = self.device.type != 'cpu' # force FP16 val during training
self.args.half = False
①取消训练时的半精度
注释 torch_utils.py中的
x['model'].half() # to FP16
②取消验证时的半精度
修改engine/validator.py中的105行为
self.args.half = self.device.type != 'cpu' # force FP16 val during training
self.args.half = False