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( save_samples(
filepath_jsonl, filepath_jsonl,
prediction_mask prediction_mask.numpy().list()
) )
i += 1 i += 1