matteo/constellationnet
matteo
/
constellationnet
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Set learning rate to 10^-3

This commit is contained in:
Mattéo Delabre 2019-12-16 10:31:29 -05:00
parent af59421dc0
commit e8e4dcec83
Signed by: matteo
GPG Key ID: AE3FBD02DC583ABB
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -61,7 +61,7 @@ scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau(
threshold=1e-8 threshold=1e-8
) )
while total_change >= 1e-4: while total_change >= 1e-3:
# Shuffle training data and convert to one-hot encoding # Shuffle training data and convert to one-hot encoding
classes_dataset = classes_ordered[torch.randperm(len(classes_ordered))] classes_dataset = classes_ordered[torch.randperm(len(classes_ordered))]
onehot_dataset = util.messages_to_onehot(classes_dataset, order) onehot_dataset = util.messages_to_onehot(classes_dataset, order)