Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'b57a03bd0d'
${ noResults }
pytorch-image-models/timm/bits
History
Ross Wightman 91ab0b6ce5
Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 		4 years ago
..
README.md First timm.bits commit, add initial abstractions, WIP updates to train, val... some of it working 4 years ago
__init__.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
avg_scalar.py Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes. 4 years ago
avg_tensor.py Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes. 4 years ago
checkpoint.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
checkpoint_manager.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
device_env.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
device_env_cuda.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
device_env_factory.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
device_env_xla.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
distributed.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
distributed_torch.py Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes. 4 years ago
grad_clip.py Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes. 4 years ago
metric.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
metric_accuracy.py Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes. 4 years ago
metric_precision_recall.py Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes. 4 years ago
monitor.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
tracker.py Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes. 4 years ago
train_cfg.py Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes. 4 years ago
train_services.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
train_setup.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
train_state.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
updater.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
updater_cuda.py Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes. 4 years ago
updater_deepspeed.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
updater_factory.py Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go... 4 years ago
updater_xla.py Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes. 4 years ago

README.md

Timm Bits

A collection of reusable components and lightweight abstractions for training and evaluating NN.

This is an early WIP with the primary goal to get up and running on TPUs first. Expect significant changes, rewrites, additions...

The current train.py and validate.py scipts are evolving to use the timm.bits components, they will also change significantly.