ColossalAI/examples/vit-b16/dataloader at 2528adc62fe41f4df3490b3c95071b85805f8d4e - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 01:03:35 +00:00

Files

History

ver217 dbe62c67b8 add an example of ViT-B/16 and remove w_norm clipping in LAMB (#29 )

2021-11-18 23:45:09 +08:00

..

__init__.py

add an example of ViT-B/16 and remove w_norm clipping in LAMB (#29 )

2021-11-18 23:45:09 +08:00

imagenet_dali_dataloader.py

add an example of ViT-B/16 and remove w_norm clipping in LAMB (#29 )

2021-11-18 23:45:09 +08:00