Searched refs:train_step_with_sens (Results 1 – 2 of 2) sorted by relevance
23 from ....train_step_wrap import train_step_with_loss_warp, train_step_with_sens62 net = train_step_with_sens(LeNet5(), sens)
83 def train_step_with_sens(network, sens): function