IrisML tasks to use timm library
Project description
irisml-tasks-timm
Timm-related tasks for IrisML pipeline.
Please visit IrisML repository for the details.
Installation
pip install irisml-tasks-timm
Tasks
create_timm_model
Create a timm model. Please use timm.list_models() method to find available model names.
class Config:
name: str
num_classes: int
pretrained: bool = False
task_type: typing.Literal['multiclass_classification', 'multilabel_classification'] = 'multiclass_classification'
create_timm_transform
Create a timm transform. The task output contains train_transform and val_transform.
class Config:
input_size: int
crop_pct: float = 0.875 # Center crop the image with this ratio.
interpolation: typing.Literal['bilinear', 'bicubic'] = 'bilinear'
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for irisml_tasks_timm-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0da0e5a54adbdf498489a209f6fb058b766c3a4f8df55b0967b3ce712b1f1861 |
|
MD5 | 8fb7784486783a9a537dda6151c082f7 |
|
BLAKE2b-256 | 12e435972d2ce937c31f1990905b353fb3d403f7e6b25f6bc466cce04fcfe049 |