How to specify prediction_key for Metric

[MaksymDel]

Hi!

When I run an example from the readme with Accuracy metric in addition to the default MSE, I get the following error:

ValueError: MetricSpec without specifiedValueError: MetricSpec without specified prediction_key requires predictions tensor or single element dict, got {'X': <tf.Tensor 'cycle_gan/transforms/G/Reshape_1:0' shape=(64, 100, 202) dtype=float32>, 'Y': <tf.Tensor 'cycle_gan/transforms/F/Reshape_1:0' shape=(64, 100, 202) dtype=float32>} requires predictions tensor or single element dict, got {'X': <tf.Tensor 'cycle_gan/transforms/G/Reshape_1:0' shape=(64, 100, 202) dtype=float32>, 'Y': <tf.Tensor 'cycle_gan/transforms/F/Reshape_1:0' shape=(64, 100, 202) dtype=float32>}

How does the correct prediction_key specification for the README example look like?

Best, Maksym

[huangsicong]

Thanks for the issue. We ran the code again and it was fine. What was the exact command you used?

Thanks, Sheldon

[aidangomez]

Closing due to inactivity. Please reopen if necessary.