MCPcopy Create free account
hub / github.com/NatLabRockies/PINNSTRIPES / force_decrease_lrm_lrw

Method force_decrease_lrm_lrw

pinn_spm_param/util/myNN.py:3365–3378  ·  view source on GitHub ↗
(
        self, lr_m, learningRateModelFinal, lr_w, learningRateWeightsFinal
    )

Source from the content-addressed store, hash-verified

3363 return
3364
3365 def force_decrease_lrm_lrw(
3366 self, lr_m, learningRateModelFinal, lr_w, learningRateWeightsFinal
3367 ):
3368 self.force_decrease_lrm(lr_m, learningRateModelFinal)
3369 lr_w_old = lr_w
3370 lr_w = max(lr_w * 0.5, learningRateWeightsFinal)
3371 filename = os.path.join(self.modelFolder, "learningRateWeights")
3372 f = open(filename, "w+")
3373 try:
3374 f.write(str(lr_w.numpy()))
3375 except:
3376 f.write(str(lr_w))
3377 f.close()
3378 return
3379
3380 def dynamic_control_lrm(self, lr_m, epoch, scheduler):
3381 lr_m_old = lr_m

Callers 1

trainMethod · 0.95

Calls 1

force_decrease_lrmMethod · 0.95

Tested by

no test coverage detected