hc/JSCC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

train.py modified

Browse Source
This commit is contained in:
chun 2023-12-23 20:47:40 +08:00
parent 3ecc36cb3c
commit 1aec37af4e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -29,7 +29,8 @@ def config_parser():
parser.add_argument('--channel', default='AWGN', type=str,
choices=['AWGN', 'Rayleigh'], help='channel type')
parser.add_argument('--saved', default='./saved', type=str, help='saved_path')
parser.add_argument('--snr_list', default=list(range(1, 19, 3)), nargs='+', help='snr_list')
parser.add_argument('--snr_list', default=['19', '13',
'7', '4', '1'], nargs='+', help='snr_list')
parser.add_argument('--ratio_list', default=['1/3',
'1/6', '1/12'], nargs='+', help='ratio_list')
parser.add_argument('--num_workers', default=0, type=int, help='num_workers')