Open Ding-Liu opened 6 years ago
All parameters are set as default values. While during the training phase, I manually adjusted the step
value to facilitate the training phase. :)
Thanks so much for your reply!
I used your code trying to reproduce the result, but my model is ~0.2dB lower than your model in terms of performance.
1) Do you mean step
is set as 5 when you did your training?
2) It seems the training datafile has the name: data/train_291_32_x234.h5
, which is slightly different from the name generated by the given Matlab code data/train_291_31_x234.h5
. Does it means the size_input
is 32 rather than 31 when your training data is generated?
Hi, @Ding-Liu
Hello! May I know the value of arguments that are used in the training command: main.py [-h] [--batchSize BATCHSIZE] [--nEpochs NEPOCHS] [--lr LR] [--step STEP] [--cuda] [--resume RESUME] [--start-epoch START_EPOCH] [--clip CLIP] [--threads THREADS] [--momentum MOMENTUM] [--weight-decay WEIGHT_DECAY] [--pretrained PRETRAINED]
For example, the values of 'nEpochs', 'step', 'batchSize' and so on... Many thanks in advance!