Replace with f-string
This commit is contained in:
@@ -131,11 +131,11 @@ class Config:
|
|||||||
or "1070" in self.gpu_name
|
or "1070" in self.gpu_name
|
||||||
or "1080" in self.gpu_name
|
or "1080" in self.gpu_name
|
||||||
):
|
):
|
||||||
logger.info("Found GPU %s, force to fp32", self.gpu_name)
|
logger.info(f"Found GPU {self.gpu_name}, force to fp32")
|
||||||
self.is_half = False
|
self.is_half = False
|
||||||
self.use_fp32_config()
|
self.use_fp32_config()
|
||||||
else:
|
else:
|
||||||
logger.info("Found GPU %s", self.gpu_name)
|
logger.info(f"Found GPU {self.gpu_name}")
|
||||||
self.gpu_mem = int(
|
self.gpu_mem = int(
|
||||||
torch.cuda.get_device_properties(i_device).total_memory
|
torch.cuda.get_device_properties(i_device).total_memory
|
||||||
/ 1024
|
/ 1024
|
||||||
|
|||||||
Reference in New Issue
Block a user