p-doom/crowd-code-dataset-0.2 huggingface dataset:
[content] /hkfs/home/project/hk-project-p0023960/tum_cte0515/Projects/jasmine/train_tokenizer.py (python)
                        gt_seq_val = val_results["gt_batch"]["videos"][0].astype(jnp.float32) / 255.0
                        recon_seq_val = val_results["recon"][0].clip(0, 1)
                        val_comparison_seq = jnp.concatenate((gt_seq_val, recon_seq_val), axis=1)
                        val_comparison_seq = einops.rearrange(
                            val_comparison_seq * 255, "t h w c -> h (t w) c"