Synchronize devices. Generate image sequence using test outputs.

print(batch['target_image'].shape) print(batch['input_image'].shape) print() print(batch['target_az'].shape)

print(batch['target_image'].shape) print(batch['input_image'].shape) print() print(batch['target_az'].shape)

Gather metrics from all devices, compute mean. Purge repeated results using data index.

Get the experiment version. Returns: The experiment version if specified else the next version.