Format code (#214)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot] 2023-05-02 20:22:08 +08:00 committed by GitHub
parent 71427575c4
commit 951989117b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions

View File

@ -53,7 +53,7 @@ class Config:
i_device = int(self.device.split(":")[-1])
self.gpu_name = torch.cuda.get_device_name(i_device)
if (
("16" in gpu_name and "V100"not in gpu_name.upper())
("16" in gpu_name and "V100" not in gpu_name.upper())
or "P40" in self.gpu_name.upper()
or "1070" in self.gpu_name
or "1080" in self.gpu_name
@ -72,12 +72,12 @@ class Config:
else:
self.gpu_name = None
self.gpu_mem = int(
torch.cuda.get_device_properties(i_device).total_memory
/ 1024
/ 1024
/ 1024
+ 0.4
)
torch.cuda.get_device_properties(i_device).total_memory
/ 1024
/ 1024
/ 1024
+ 0.4
)
if self.gpu_mem <= 4:
with open("trainset_preprocess_pipeline_print.py", "r") as f:
strr = f.read().replace("3.7", "3.0")
@ -90,7 +90,7 @@ class Config:
print("没有发现支持的N卡, 使用CPU进行推理")
self.device = "cpu"
self.is_half = True
if self.n_cpu == 0:
self.n_cpu = cpu_count()