pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

timm bits

Open rwightman opened this issue 3 years ago • 1 comments

The branch associated with this PR is for development of timm bits -- a significant update and reorganizaiton of timm's training scripts and associated modules. It works w/ TPUs (PyTorch XLA) and GPUs (PyTorch) and possibly DeepSpeed.

This PR is still a long ways from being merged. Having the branch as a PR helps w/ visibility of the ongoing work.

rwightman avatar Aug 13 '21 20:08 rwightman