Searched refs:losses_dict (Results 1 – 1 of 1) sorted by relevance
310 losses_dict = {}312 losses_dict[layer] = {'losses': layer._losses[:],317 return losses_dict320 def _restore_layer_losses(losses_dict): argument321 for layer in losses_dict:323 layer._losses = losses_dict[layer]['losses']324 layer._eager_losses = losses_dict[layer]['eager_losses']