Browse Source

更新 'train_onestep.py'

master
Mack189 1 month ago
parent
commit
0538ae0c08
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      train_onestep.py

+ 2
- 2
train_onestep.py View File

@@ -64,14 +64,14 @@ parser.add_argument('--valid', type=str, default="/home/work/user-job-dir/inputs
help='Sample rate of audio file')
parser.add_argument('--test', type=str, default="/home/work/user-job-dir/inputs/data_json/tr",
help='Sample rate of audio file')
parser.add_argument('--lr', type=float, default=1e-4,
parser.add_argument('--lr', type=float, default=8e-5,
help='Sample rate of audio file')
parser.add_argument('--beta2', type=float, default=0.999,
help='Sample rate of audio file')
parser.add_argument('--snapshots', type=int, default=1, help='Snapshots')
parser.add_argument('--prefix', default='tpami_residual_filter8', help='Location to save checkpoint models')
parser.add_argument('--model_type', type=str, default='swave')
parser.add_argument('--pretrained_sr', default='9_gdprnn_1.ckpt', help='sr pretrained base model')
parser.add_argument('--pretrained_sr', default='20_gdprnn.ckpt', help='sr pretrained base model')

#asd
def train(trainoneStep, data, train_dir, obs_train_url, args):


Loading…
Cancel
Save