pytorch-image-models

Commit Graph

Author	SHA1	Message	Date
Ross Wightman	4f338556d8	Fixes and improvements for metrics, tfds parser, loader / transform handling * add back ability to create transform with loader * change 'samples' -> 'examples' for tfds wrapper to match tfds naming * add support for specifying feature names for input and target in tfds wrapper * add class_to_idx for image classification datasets in tfds wrapper * add accumulate_type to avg meters and metrics to allow float32 or float64 accumulation control with lower prec data * minor cleanup, log output rate prev and avg	4 years ago
Ross Wightman	aa92d7b1c5	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

Author

SHA1

Message

Date

Ross Wightman

4f338556d8

Fixes and improvements for metrics, tfds parser, loader / transform handling

* add back ability to create transform with loader
* change 'samples' -> 'examples' for tfds wrapper to match tfds naming
* add support for specifying feature names for input and target in tfds wrapper
* add class_to_idx for image classification datasets in tfds wrapper
* add accumulate_type to avg meters and metrics to allow float32 or float64 accumulation control with lower prec data
* minor cleanup, log output rate prev and avg

Ross Wightman

aa92d7b1c5

Major timm.bits update. Updater and DeviceEnv now dataclasses, after_step closure used, metrics base impl w/ distributed reduce, many tweaks/fixes.

2 Commits (c639a86c67d449b39a1c08542f5eb06ced84ca84)