dlr: tf.Tensor → numpy array → list → json → disk

This commit is contained in:
Starbeamrainbowlabs 2023-03-10 17:34:49 +00:00
parent b287970032
commit 5e79825292
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -341,7 +341,7 @@ def plot_predictions(filepath, input_items, colormap, model):
save_samples(
filepath_jsonl,
prediction_mask
prediction_mask.numpy().list()
)
i += 1