def assert_equal_metrics(m1, m2): assert_tensors(m1.pop('input_layer', None), m2.pop('input_layer', None)) assert_tensors(m1.pop('output_layer', None), m2.pop('output_layer', None)) assert_equal_dict(m1, m2)
def assert_equal_losses(l1, l2): assert_tensors(l1.pop('input_layer', None), l2.pop('input_layer', None)) assert_tensors(l1.pop('output_layer', None), l2.pop('output_layer', None)) assert_equal_dict(l1, l2)