Skip to content

Commit 83ff684

Browse files
author
Pedro Fonseca Esteves
committed
fixed typo
1 parent aaf9147 commit 83ff684

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ratsql/commands/train.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def train(self, config, modeldir):
139139
last_step = saver.restore(modeldir, map_location=self.device)
140140

141141
#lr fix to not break scheduler when loading from checkpoint
142-
lr_scheduler.param_groups = optimizer.param_group
142+
lr_scheduler.param_groups = optimizer.param_groups
143143

144144
if "pretrain" in config and last_step == 0:
145145
pretrain_config = config["pretrain"]

0 commit comments

Comments
 (0)