def main():
sess = K.get_session()
sess = tf_debug.LocalCLIDebugWrapperSession(sess)
K.set_session(sess)
sess.add_tensor_filter("has_inf_or_nan", tf_debug.has_inf_or_nan)
style = StyleTransfer(stop_layer='block1_conv1')
style.train("content", "style")
