Seems to be an indexing problem when running experiments with the functional API. From the mailing list:

I'm running with a simple trainable function (so no Trainable sub-classing) but I still seem to have every trial run have '2' as the iteration count, e.g. TERMINATED [pid=3942], 327 s, 2 iter <--- how is this possible? Surely there's only 'one' iteration per trial to be done if I'm not doing any checkpointing etc?