File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -220,7 +220,7 @@ def forward(self, x):
220
220
#
221
221
222
222
optimizer = optim .Adam (model .parameters (), lr = 0.01 , weight_decay = 0.0001 )
223
- scheduler = optim .lr_scheduler .StepLR (optimizer , step_size = 30 , gamma = 0.1 )
223
+ scheduler = optim .lr_scheduler .StepLR (optimizer , step_size = 20 , gamma = 0.1 )
224
224
225
225
226
226
######################################################################
@@ -281,7 +281,7 @@ def test(model, epoch):
281
281
#
282
282
283
283
log_interval = 20
284
- for epoch in range (1 , 61 ):
284
+ for epoch in range (1 , 41 ):
285
285
if epoch == 31 :
286
286
print ("First round of training complete. Setting learn rate to 0.001." )
287
287
scheduler .step ()
@@ -302,6 +302,6 @@ def test(model, epoch):
302
302
# For more advanced audio applications, such as speech recognition,
303
303
# recurrent neural networks (RNNs) are commonly used. There are also other
304
304
# data preprocessing methods, such as finding the mel frequency cepstral
305
- # coefficients (MCFF ), that can reduce the size of the dataset.
305
+ # coefficients (MFCC ), that can reduce the size of the dataset.
306
306
#
307
307
You can’t perform that action at this time.
0 commit comments