fix config
This commit is contained in:
parent
b3d7075ba4
commit
d2ea3f1930
@ -17,9 +17,9 @@ def use_fp32_config():
|
|||||||
strr = f.read().replace("true", "false")
|
strr = f.read().replace("true", "false")
|
||||||
with open(f"configs/{config_file}", "w") as f:
|
with open(f"configs/{config_file}", "w") as f:
|
||||||
f.write(strr)
|
f.write(strr)
|
||||||
with open("trainset_preprocess_pipeline_print.py", "r") as f:
|
with open("infer/modules/train/preprocess.py", "r") as f:
|
||||||
strr = f.read().replace("3.7", "3.0")
|
strr = f.read().replace("3.7", "3.0")
|
||||||
with open("trainset_preprocess_pipeline_print.py", "w") as f:
|
with open("infer/modules/train/preprocess.py", "w") as f:
|
||||||
f.write(strr)
|
f.write(strr)
|
||||||
|
|
||||||
|
|
||||||
@ -110,9 +110,9 @@ class Config:
|
|||||||
+ 0.4
|
+ 0.4
|
||||||
)
|
)
|
||||||
if self.gpu_mem <= 4:
|
if self.gpu_mem <= 4:
|
||||||
with open("trainset_preprocess_pipeline_print.py", "r") as f:
|
with open("infer/modules/train/preprocess.py", "r") as f:
|
||||||
strr = f.read().replace("3.7", "3.0")
|
strr = f.read().replace("3.7", "3.0")
|
||||||
with open("trainset_preprocess_pipeline_print.py", "w") as f:
|
with open("infer/modules/train/preprocess.py", "w") as f:
|
||||||
f.write(strr)
|
f.write(strr)
|
||||||
elif self.has_mps():
|
elif self.has_mps():
|
||||||
print("No supported Nvidia GPU found")
|
print("No supported Nvidia GPU found")
|
||||||
|
Loading…
Reference in New Issue
Block a user