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