barian0517/RCV
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore(format): run black on dev (#1717)

Browse Source 
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot] 2024-01-14 16:43:14 +09:00 committed by GitHub
parent bb4d185a57
commit 4f7b6b2b9e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
configs/config.py
View File

@ -158,7 +158,8 @@ class Config:
/ 1024 / 1024
+ 0.4 + 0.4
) )
if self.gpu_mem <= 4: self.preprocess_per = 3.0 if self.gpu_mem <= 4:
self.preprocess_per = 3.0
elif self.has_mps(): elif self.has_mps():
logger.info("No supported Nvidia GPU found") logger.info("No supported Nvidia GPU found")
self.device = self.instead = "mps" self.device = self.instead = "mps"
@ -241,5 +242,8 @@ class Config:
) )
except: except:
pass pass
logger.info("Half-precision floating-point: %s, device: %s" % (self.is_half, self.device)) logger.info(
"Half-precision floating-point: %s, device: %s"
% (self.is_half, self.device)
)
return x_pad, x_query, x_center, x_max return x_pad, x_query, x_center, x_max

1
infer/modules/train/preprocess.py
View File

@ -31,6 +31,7 @@ def println(strr):
f.write("%s\n" % strr) f.write("%s\n" % strr)
f.flush() f.flush()
class PreProcess: class PreProcess:
def __init__(self, sr, exp_dir, per=3.7): def __init__(self, sr, exp_dir, per=3.7):
self.slicer = Slicer( self.slicer = Slicer(