Searched refs:get_small_subclass_mlp (Results 1 – 2 of 2) sorted by relevance
514 def get_small_subclass_mlp(num_hidden, num_classes): function526 return get_small_subclass_mlp(num_hidden, num_classes)
73 self.subclassed_model = testing_utils.get_small_subclass_mlp(1, 2)