Workaround for this crash on Tensorflow 2.13:
Could not load library libcublasLt.so.12. Error: libcublasLt.so.12: cannot open shared object file: No such file or directory
it should tie into Tensorflow's logging just fine so long as it's the first callback in the queue.
***** TEST SCRIPT *****
model = tf.keras.Sequential([
tf.keras.layers.Dense(64, activation='relu', input_shape=(10,)),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dense(1)
])
model.compile(optimizer='adam', loss='mse', metrics=['mae'])
X = np.random.random((100, 10))
y = np.random.random((100, 1))
split = 80
X_train, X_val = X[:split], X[split:]
y_train, y_val = y[:split], y[split:]
train_dataset = tf.data.Dataset.from_tensor_slices((X_train, y_train)).batch(10)
val_dataset = tf.data.Dataset.from_tensor_slices((X_val, y_val)).batch(10)
history = model.fit(train_dataset,
epochs=10,
validation_data=val_dataset,
callbacks=[
CallbackExtraValidation({
"test": val_dataset
}, verbose=0),
tf.keras.callbacks.CSVLogger("/dev/stdout", separator="\t")
],
verbose=0
)
print(f"DEBUG history {history}")
This removes the train_percentage argument
TODO: map this forwards to enable support in deeplabv3_plus_test_rainfall
...thinking about it, it's really not a test now, is it? Updating the filename would be such a /hassle/ though....