import tensorflow as tf import tensorflow.keras.layers as l import generic as g m = tf.keras.models.Sequential([ l.Input(shape = (2,)), l.Dense(1000, activation = "relu"), l.Dense(1) ]) #g.train_generic(m, "ff2") g.verify_generic(m, "ff2")