Update DALI pipeline for ViT

NVIDIA / JAX-Toolbox

JAX-Toolbox

Apache License 2.0

210 stars 35 forks source link

Closed awolant closed 1 week ago

awolant commented 4 weeks ago

This PR changes the DALI pipeline for ViT. It is no longer necessary to combine two pipelines together and this PR applies this to ViT pipeline.

terrykong commented 3 weeks ago

@ashors1 to review and also run internal CI on this change