Report precision

This commit is contained in:
cmdr2 2022-12-24 21:44:42 +05:30
parent 01320ac735
commit a2b55c0df7

View File

@ -115,7 +115,7 @@ def device_init(context, device):
# Apply force_full_precision now before models are loaded.
context.half_precision = False
log.info(f'Setting {device} as active')
log.info(f'Setting {device} as active, with precision: {"half" if context.half_precision else "full"}')
torch.cuda.device(device)
return