This commit is contained in:
Ftps
2023-08-28 16:08:31 +09:00
parent 3c7f1f1407
commit 58e32b6def
55 changed files with 237 additions and 169 deletions

View File

@@ -1,9 +1,10 @@
import os
import argparse
import os
import sys
import torch
from multiprocessing import cpu_count
import torch
def use_fp32_config():
for config_file in [
@@ -198,6 +199,3 @@ class Config:
except:
pass
return x_pad, x_query, x_center, x_max
defaultconfig = Config()