AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update train.py

Browse Source
This commit is contained in:
jingyong hou 2021-11-19 17:18:58 +08:00
parent 8cd12edfed
commit 17cd3c47e5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
kws/bin/train.py
View File

@ -103,9 +103,9 @@ def main():
logging.basicConfig(level=logging.DEBUG, logging.basicConfig(level=logging.DEBUG,
format='%(asctime)s %(levelname)s %(message)s') format='%(asctime)s %(levelname)s %(message)s')
os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu) os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu)
set_mannul_seed(args.gpu)
# Set random seed # Set random seed
torch.manual_seed(777) set_mannul_seed(args.seed)
print(args) print(args)
with open(args.config, 'r') as fin: with open(args.config, 'r') as fin:
configs = yaml.load(fin, Loader=yaml.FullLoader) configs = yaml.load(fin, Loader=yaml.FullLoader)