An AutoML Library made with Optuna and PyTorch Lightning

Overview

logo

An AutoML Library made with Optuna and PyTorch Lightning

Maintainability Rating Reliability Rating Security Rating codecov Documentation Status DOI Slack

PyPI - Python Version PyPI version Downloads Downloads license

Installation

Recommended

pip install -U gradsflow

From source

pip install git+https://github.com/gradsflow/[email protected]

Highlights

What is Gradsflow?

Gradsflow is based on Optuna and PyTorch Lightning โšก๏ธ . It leverages PyTorch Lightning Flash so that you don't have to write any PyTorch or Optuna code for model building or hyperparameter tuning ๐Ÿš€

Although you might want to train a custom model and search hyperparameters, You can easily integrate any PyTorch/Lightning Flash Model with Gradsflow AutoModel โœจ

  • gradsflow.core: Core defines the building blocks of AutoML tasks.

  • gradsflow.taskauto: AutoTasks defines different ML/DL tasks which is provided by Gradsflow AutoML API.

๐Ÿ“‘ Check out notebooks examples.

๐Ÿ’ฌ Join the Slack group to chat with us.

๐Ÿค— Contribute

Contributions of any kind are welcome. Please check the Contributing Guidelines before contributing.

Code Of Conduct

We pledge to act and interact in ways that contribute to an open, welcoming, diverse, inclusive, and healthy community.

Read full Contributor Covenant Code of Conduct

Acknowledgement

Gradsflow is built with help of Optuna and PyTorch Lightning ๐Ÿ’œ

Comments
  • Tensorboard callbacks

    Tensorboard callbacks

    Changes

    Fixes #123

    Type of change

    • [ ] ๐Ÿ“š Documentation Update
    • [ ] ๐Ÿงช Tests Cases
    • [ ] ๐Ÿž Bug fix (non-breaking change which fixes an issue)
    • [x] ๐Ÿ”ฌ New feature (non-breaking change which adds functionality)
    • [ ] ๐Ÿšจ Breaking change (fix or feature that would cause existing functionality to not work as expected)
    • [ ] ๐Ÿ“ This change requires a documentation update

    Checklist

    • [x] My code follows the style guidelines of this project
    • [x] I have performed a self-review of my own code
    • [ ] I have commented my code, particularly in hard-to-understand areas
    • [ ] I have made corresponding changes to the documentation
    • [x] My changes generate no new warnings
    • [x] Did you update CHANGELOG (docs/CHANGELOG.md) in case of a major change?

    Solves issue #123 I have added basic tensorboard callback functionality which include: logging the loss and accuracy for both training and validation. Is there any other functionality of tensorboard that I need to add @aniketmaurya

    documentation enhancement 
    opened by arvindmuralie77 10
  • (PYL-W0613) Function contains unused argument

    (PYL-W0613) Function contains unused argument

    Description

    An unused argument can lead to confusions. It should be removed. If this variable is necessary, name the variable _ or start the name with unused or _unused.

    Occurrences

    There are 5 occurrences of this issue in the repository.

    See all occurrences on DeepSource โ†’ deepsource.io/gh/gradsflow/gradsflow/issue/PYL-W0613/occurrences/

    no-issue-activity 
    opened by aniketmaurya 8
  • Add a conda installation option

    Add a conda installation option

    I believe, adding a conda installation option for gradsflow will be helpful for growth and adoption of the library. I have started the work on it already. Once the :bulb: PR gets approved and merged, you will have gradsflow on conda-forge.

    conda install -c conda-forge gradsflow
    

    :no_entry: :fire: Roadblock to conda-forge packaging

    However, there seems to be a problem: it appears that this library is somewhat tightly coupled with comet_ml (which has a proprietary license -- NOT OpenSource). If you could work on making this a weak coupling, or better yet make comet_ml optional (even for tests), that would allow us to make gradsflow available on conda-forge.

    opened by sugatoray 8
  • implement callbacks

    implement callbacks

    ๐Ÿš˜ Callback Roadmap

    • [x] #120 - @aniketmaurya #121
    • [ ] #122 - @aniketmaurya https://github.com/gradsflow/gradsflow/pull/151
    • [x] Tensorboard #123
    • [x] Comet #125
    • [x] WandB Callback #124
    • [x] CSV Logger - @aniketmaurya #116
    enhancement help wanted hacktoberfest no-issue-activity 
    opened by aniketmaurya 7
  • ๐Ÿš€ CLI

    ๐Ÿš€ CLI

    Changes

    New feature: AutoML Training with CLI

    Type of change

    • [ ] ๐Ÿ“š Documentation Update
    • [ ] ๐Ÿงช Tests Cases
    • [ ] ๐Ÿž Bug fix (non-breaking change which fixes an issue)
    • [x] ๐Ÿ”ฌ New feature (non-breaking change which adds functionality)
    • [ ] ๐Ÿšจ Breaking change (fix or feature that would cause existing functionality to not work as expected)
    • [ ] ๐Ÿ“ This change requires a documentation update

    Checklist

    • [ ] My code follows the style guidelines of this project
    • [ ] I have performed a self-review of my own code
    • [ ] I have commented my code, particularly in hard-to-understand areas
    • [ ] I have made corresponding changes to the documentation
    • [ ] My changes generate no new warnings
    • [ ] Did you update CHANGELOG in case of a major change?
    enhancement API design no-pr-activity 
    opened by aniketmaurya 5
  • Argparse support added.

    Argparse support added.

    Feature Request

    Type of change

    • [X] ๐Ÿ”ฌ New feature (non-breaking change which adds functionality)

    Hi @aniketmaurya,

    I added the argparser property for the image_classifier.py file.

    example no-pr-activity 
    opened by kadirnar 4
  • ๐Ÿ”ฅ Support Schedulers

    ๐Ÿ”ฅ Support Schedulers

    Changes

    Fixes #88

    Type of change

    • [ ] ๐Ÿ“š Documentation Update
    • [ ] ๐Ÿงช Tests Cases
    • [ ] ๐Ÿž Bug fix (non-breaking change which fixes an issue)
    • [x] ๐Ÿ”ฌ New feature (non-breaking change which adds functionality)
    • [ ] ๐Ÿšจ Breaking change (fix or feature that would cause existing functionality to not work as expected)
    • [ ] ๐Ÿ“ This change requires a documentation update

    Checklist

    • [x] My code follows the style guidelines of this project
    • [x] I have performed a self-review of my own code
    • [x] I have commented my code, particularly in hard-to-understand areas
    • [x] I have made corresponding changes to the documentation
    • [x] My changes generate no new warnings
    • [ ] Did you update CHANGELOG in case of a major change?
    enhancement test no-pr-activity 
    opened by aniketmaurya 4
  • โ˜„๏ธ comet integration

    โ˜„๏ธ comet integration

    Changes

    Fixes #125

    Type of change

    • [ ] ๐Ÿ“š Documentation Update
    • [ ] ๐Ÿงช Tests Cases
    • [ ] ๐Ÿž Bug fix (non-breaking change which fixes an issue)
    • [x] ๐Ÿ”ฌ New feature (non-breaking change which adds functionality)
    • [ ] ๐Ÿšจ Breaking change (fix or feature that would cause existing functionality to not work as expected)
    • [ ] ๐Ÿ“ This change requires a documentation update

    Checklist

    • [x] My code follows the style guidelines of this project
    • [x] I have performed a self-review of my own code
    • [x] I have commented my code, particularly in hard-to-understand areas
    • [x] I have made corresponding changes to the documentation
    • [x] My changes generate no new warnings
    • [ ] Did you update CHANGELOG in case of a major change?
    enhancement test example API design 
    opened by aniketmaurya 4
  • Automatic Task Selection

    Automatic Task Selection

    Is your feature request related to a problem? Please describe.

    Create Tasks directly from AutoClassifier instead of explicitly calling AutoImageClassification or AutoTextSummarization

    Describe the solution you'd like

    model=AutoClassification(datamodule, data_type="image")  # expected `data_type`-> image, text, infer
    model.hp_tune()
    
    enhancement good first issue help wanted 
    opened by aniketmaurya 4
  • migrate to ray_tune

    migrate to ray_tune

    Changes

    Fixes #35

    Type of change

    • [ ] Documentation Update
    • [ ] Bug fix (non-breaking change which fixes an issue)
    • [x] New feature (non-breaking change which adds functionality)
    • [x] Breaking change (fix or feature that would cause existing functionality to not work as expected)
    • [x] This change requires a documentation update

    Checklist

    • [x] My code follows the style guidelines of this project
    • [x] I have performed a self-review of my own code
    • [x] I have commented my code, particularly in hard-to-understand areas
    • [x] I have made corresponding changes to the documentation
    • [x] My changes generate no new warnings
    documentation enhancement test example API design 
    opened by aniketmaurya 4
  • Adding example notebook for AutoSummarization

    Adding example notebook for AutoSummarization

    Changes

    Fixes #5 (issue)

    Type of change

    • [x] New feature (non-breaking change which adds functionality)

    Checklist

    • [x] My code follows the style guidelines of this project
    • [x] I have performed a self-review of my own code
    • [x] I have commented my code, particularly in hard-to-understand areas
    • [x] I have made corresponding changes to the documentation
    • [x] My changes generate no new warnings
    example 
    opened by gagan3012 4
Releases(v0.0.8.post1)
  • v0.0.8.post1(May 18, 2022)

    What's Changed

    • ๐ŸŒŸ format docs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/170
    • [Snyk] Fix for 3 vulnerabilities by @snyk-bot in https://github.com/gradsflow/gradsflow/pull/171
    • Tensorboard callbacks by @arvindmuralie77 in https://github.com/gradsflow/gradsflow/pull/173
    • update docs :memo: by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/175
    • anti pattern fixes from deepsource by @skp-github in https://github.com/gradsflow/gradsflow/pull/176
    • [pre-commit.ci] pre-commit suggestions by @pre-commit-ci in https://github.com/gradsflow/gradsflow/pull/177
    • Demo & fixes by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/180
    • Upgrade deps by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/181
    • ๐Ÿ› minor bug fixes and reformat by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/182
    • fix examples and Flash trainer by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/183
    • refactor apis by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/184
    • Fix examples by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/185

    New Contributors

    • @snyk-bot made their first contribution in https://github.com/gradsflow/gradsflow/pull/171
    • @arvindmuralie77 made their first contribution in https://github.com/gradsflow/gradsflow/pull/173
    • @skp-github made their first contribution in https://github.com/gradsflow/gradsflow/pull/176

    Full Changelog: https://github.com/gradsflow/gradsflow/compare/v0.0.8...v0.0.8.post1

    Source code(tar.gz)
    Source code(zip)
  • v0.0.8(Jan 14, 2022)

    What's Changed

    • ๐Ÿคฉ refactor core by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/136
    • cleanup APIs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/137
    • added conda installation instruction by @sugatoray in https://github.com/gradsflow/gradsflow/pull/144
    • recursively exclude tests folder and its contents by @sugatoray in https://github.com/gradsflow/gradsflow/pull/141
    • add model.save test by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/147
    • remove redundant to_item by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/152
    • refactor Tracker by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/153
    • Change methods not using its bound instance to staticmethods by @deepsource-autofix in https://github.com/gradsflow/gradsflow/pull/156
    • refactor metrics by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/159
    • add dataoader length by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/160
    • fix model checkpoint folder not found by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/162
    • Fix metrics update by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/163
    • Replace multiple == checks with in by @deepsource-autofix in https://github.com/gradsflow/gradsflow/pull/167
    • increment current_epoch after each epoch by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/169
    • Wandb Implementation by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/168

    Full Changelog: https://github.com/gradsflow/gradsflow/compare/v0.0.7...v0.0.8

    Source code(tar.gz)
    Source code(zip)
  • v0.0.8.dev1(Jan 12, 2022)

    What's Changed

    • optional pl dependency by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/133
    • minor fixes by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/134
    • ๐Ÿ“š update example by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/135
    • ๐Ÿคฉ refactor core by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/136
    • cleanup APIs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/137
    • remove dependencies by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/139
    • added conda installation instruction by @sugatoray in https://github.com/gradsflow/gradsflow/pull/144
    • recursively exclude tests folder and its contents by @sugatoray in https://github.com/gradsflow/gradsflow/pull/141
    • pin flash version to 0.5.1 by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/143
    • refactor backend by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/146
    • Remove flit by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/148
    • add model.save test by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/147
    • Update main.yml by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/149
    • remove redundant to_item by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/152
    • [pre-commit.ci] pre-commit suggestions by @pre-commit-ci in https://github.com/gradsflow/gradsflow/pull/155
    • refactor Tracker by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/153
    • Change methods not using its bound instance to staticmethods by @deepsource-autofix in https://github.com/gradsflow/gradsflow/pull/156
    • ๐Ÿ“ fix documentation & examples by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/158
    • refactor metrics by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/159
    • add dataoader length by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/160
    • fix model checkpoint folder not found by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/162
    • Fix metrics update by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/163

    Full Changelog: https://github.com/gradsflow/gradsflow/compare/v0.0.7...v0.0.8.dev1

    Source code(tar.gz)
    Source code(zip)
  • v0.0.7.post2(Dec 11, 2021)

    What's Changed

    • optional pl dependency by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/133
    • minor fixes by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/134
    • ๐Ÿ“š update example by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/135
    • ๐Ÿคฉ refactor core by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/136
    • cleanup APIs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/137

    Full Changelog: https://github.com/gradsflow/gradsflow/compare/v0.0.7...v0.0.7.post2

    Source code(tar.gz)
    Source code(zip)
  • v0.0.7(Nov 26, 2021)

    Highlights

    • โ˜„๏ธ comet integration #129
    • add model checkpoint callback #121
    • ๐Ÿ“ add csv logger #116
    • ๐Ÿš€ add train_eval_callback #111
    • ๐Ÿช„ add Average Meter #109
    • fix device issue in metric calculation PR #106

    What's Changed

    • [pre-commit.ci] pre-commit suggestions by @pre-commit-ci in https://github.com/gradsflow/gradsflow/pull/98
    • clean docs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/100
    • fix optimizer by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/102
    • fix device issue in metric calculation by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/106
    • โœจ refactor tuner by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/107
    • ๐Ÿ“š add example & better intro by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/108
    • ๐Ÿช„ add Average Meter by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/109
    • Fixes metrics device handling by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/110
    • ๐Ÿš€ add train_eval_callback by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/111
    • refactor training callback by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/112
    • ๐Ÿ”ฅData refactor by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/113
    • ๐Ÿš€ Bump test coverage by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/114
    • fix live display error on multiple runs by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/115
    • ๐Ÿ“ add csv logger by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/116
    • โœจrefactor base model by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/118
    • โ˜˜๏ธ add codecarbon callback by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/119
    • add model checkpoint callback by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/121
    • fix loss compile by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/126
    • add requirement decorator by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/128
    • Refactor callback runner by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/130
    • โ˜„๏ธ comet integration by @aniketmaurya in https://github.com/gradsflow/gradsflow/pull/129

    New Contributors

    • @pre-commit-ci made their first contribution in https://github.com/gradsflow/gradsflow/pull/98

    Full Changelog: https://github.com/gradsflow/gradsflow/compare/v0.0.6...v0.0.7

    Source code(tar.gz)
    Source code(zip)
  • v0.0.6(Oct 4, 2021)

    0.0.6

    • ๐ŸŽ‰ Revamp Callbacks and Training #94
    • โœจ refactor data handling ๐Ÿ“ docs update. PR #91
    • integrate torchmetrics. PR #80
    • callbacks & ๐Ÿค‘ ProgressCallback. PR #76
    • ๐Ÿ”ฅ Add AutoModel Tuner. PR #74
    • refactor APIs - Simplify API & add model.compile(...). PR #73
    • ๐Ÿค— integrate HF Accelerator. PR #71
    Source code(tar.gz)
    Source code(zip)
  • v0.0.5(Sep 26, 2021)

    What is New?

    • Keras style Model Training API ๐ŸŽ‰
    • Remote Dataset Loader - create dataloader from any cloud Bucket โ˜๏ธ
    • Datagenerator for Image Classification (more to come soon...)

    ๐Ÿ“š Documentation & Examples

    To install $ pip install -U gradsflow

    Release Notes

    • ๐Ÿ”ฅ Add custom training loop with model.fit. PR #63 Done by @aniketmaurya
    • โ˜๏ธ Add ray.data - remote dataset loader. PR #61 Done by @aniketmaurya
    • ๐ŸŽ‰ Add AutoDataset - Encapsulate datamodule and dataloaders. PR #59 Done by @aniketmaurya
    • ๐ŸŒŸ Add Autotask feature. PR #54 Done by @gagan3012
    • โœจ Add AutoTrainer to support plain torch training loop and other torch frameworks. PR #53
    Source code(tar.gz)
    Source code(zip)
  • v0.0.5a0(Sep 25, 2021)

    Release Notes

    ๐Ÿ”ฅ Add custom training loop with model.fit. PR #63 Done by @aniketmaurya โ˜๏ธ Add ray.data - remote dataset loader. PR #61 Done by @aniketmaurya ๐ŸŽ‰ Add AutoDataset - Encapsulate datamodule and dataloaders. PR #59 Done by @aniketmaurya ๐ŸŒŸ Add Autotask feature. PR #54 Done by @gagan3012 โœจ Add AutoTrainer to support plain torch training loop and other torch frameworks. PR #53

    $ pip install gradsflow==0.0.5a0

    ๐Ÿ“š Get started with Documentation

    Source code(tar.gz)
    Source code(zip)
  • v0.0.4(Sep 3, 2021)

    Latest Changes

    • fix best checkpoints model loading #52
    • ๐Ÿš€ feature/fix train arguments docs #44
    • Publish Python ๐Ÿ distributions ๐Ÿ“ฆ to PyPI #42
    Source code(tar.gz)
    Source code(zip)
  • v0.0.3(Aug 30, 2021)

    0.0.3

    Latest Changes

    Get Started Now

    ๐Ÿ“š Documentation: docs.gradsflow.com

    $ pip install -U gradsflow

    Example

    from gradsflow import AutoImageClassifier
    
    from flash.core.data.utils import download_data
    from flash.image import ImageClassificationData
    
    
    data_dir = "/Users/aniket/personal/gradsflow/gradsflow/data/"
    download_data("https://pl-flash-data.s3.amazonaws.com/hymenoptera_data.zip", data_dir)
    datamodule = ImageClassificationData.from_folders(
        train_folder=f"{data_dir}/hymenoptera_data/train/",
        val_folder=f"{data_dir}/hymenoptera_data/val/",
    )
    
    model = AutoImageClassifier(
        datamodule,
        max_epochs=2,
        n_trials=4,
        optimization_metric="val_accuracy",
        timeout=50,
    )
    
    print("AutoImageClassifier initialised!")
    model.hp_tune(gpu=1)
    
    Source code(tar.gz)
    Source code(zip)
  • v0.0.3a2(Aug 29, 2021)

    • migrate to ray_tune ๐ŸŒŸ. Read more here. PR #36 by @aniketmaurya.
    • render jupyter notebooks in documentation. PR #38 by @aniketmaurya.

    $ pip install -U gradsflow==0.0.3a2

    ๐Ÿ“ Read more on documentation

    Source code(tar.gz)
    Source code(zip)
  • v0.0.3a1(Aug 26, 2021)

  • v0.0.2(Aug 26, 2021)

  • v0.0.1(Aug 25, 2021)

    Release Notes

    0.0.1

    What is Gradsflow?

    Gradsflow is based on Optuna and PyTorch Lightning โšก๏ธ. It leverages PyTorch Lightning Flash so that you don't have to write any PyTorch or Optuna code for model building or hyperparameter tuning ๐Ÿš€

    Although you might want to train a custom model and search hyperparameters, You can easily integrate any PyTorch/Lightning Flash Model with Gradsflow AutoModel โœจ

    gradsflow.core: Core defines the building blocks of AutoML tasks.

    gradsflow.taskauto: AutoTasks defines different ML/DL tasks which is provided by Gradsflow AutoML API.

    Image classification example

        from flash.core.data.utils import download_data
        from flash.image import ImageClassificationData
    
        from gradsflow import AutoImageClassifier
    
        # 1. Create the DataModule
        download_data("https://pl-flash-data.s3.amazonaws.com/hymenoptera_data.zip", "./data")
    
        datamodule = ImageClassificationData.from_folders(
            train_folder="data/hymenoptera_data/train/",
            val_folder="data/hymenoptera_data/val/",
        )
    
        suggested_conf = dict(
            optimizers=["adam", "sgd"],
            lr=(5e-4, 1e-3),
        )
        model = AutoImageClassifier(datamodule,
                                    suggested_conf=suggested_conf,
                                    max_epochs=10,
                                    optimization_metric="val_accuracy",
                                    timeout=300)
        model.hp_tune()
    
    
    Source code(tar.gz)
    Source code(zip)
  • v0.0.1b1(Aug 24, 2021)

  • 0.0.1a1(Aug 22, 2021)

    • Added AutoImageClassification
    • Added AutoTextClassification

    Example

    model = AutoImageClassifier(datamodule,
                                suggested_backbones=['ssl_resnet18'],
                                suggested_conf=suggested_conf,
                                max_epochs=1,
                                optimization_metric="val_accuracy",
                                timeout=30)
    
    print("AutoImageClassifier initialised!")
    model.hp_tune()
    
    Source code(tar.gz)
    Source code(zip)
Owner
GradsFlow
No Code Artificial Intelligence
GradsFlow
Tiny Kinetics-400 for test

Kinetics-400่ฟทไฝ ๆ•ฐๆฎ้›† English | ็ฎ€ไฝ“ไธญๆ–‡ ่ฏฅๆ•ฐๆฎ้›†ๆ—จๅœจ่งฃๅ†ณ็š„้—ฎ้ข˜๏ผšๅ‚็…งKinetics-400ๆ•ฐๆฎๆ ผๅผ๏ผŒ่ฎญ็ปƒๅŸบไบŽ่‡ชๅทฑๆ•ฐๆฎ็š„่ง†้ข‘็†่งฃๆจกๅž‹ใ€‚ ๆ•ฐๆฎ้›†ไป‹็ป Kinetics-400ๆ˜ฏ่ง†้ข‘้ข†ๅŸŸbenchmarkๅธธ็”จๆ•ฐๆฎ้›†๏ผŒ่ฏฆ็ป†ไป‹็ปๅฏไปฅๅ‚่€ƒๅ…ถๅฎ˜ๆ–น็ฝ‘็ซ™Kineticsใ€‚ๆ•ดไธชๆ•ฐๆฎ้›†ๅŒ…ๅซ40

38 Jan 06, 2023
Bayesian Generative Adversarial Networks in Tensorflow

Bayesian Generative Adversarial Networks in Tensorflow This repository contains the Tensorflow implementation of the Bayesian GAN by Yunus Saatchi and

Andrew Gordon Wilson 1k Nov 29, 2022
Scripts for training an AI to play the endless runner Subway Surfers using a supervised machine learning approach by imitation and a convolutional neural network (CNN) for image classification

About subwAI subwAI - a project for training an AI to play the endless runner Subway Surfers using a supervised machine learning approach by imitation

82 Jan 01, 2023
GNN-based Recommendation Benchma

GRecX A Fair Benchmark for GNN-based Recommendation Preliminary Comparison DiffNet-Yelp dataset (featureless) Algo 73 Oct 17, 2022

FANet - Real-time Semantic Segmentation with Fast Attention

FANet Real-time Semantic Segmentation with Fast Attention Ping Hu, Federico Perazzi, Fabian Caba Heilbron, Oliver Wang, Zhe Lin, Kate Saenko , Stan Sc

Ping Hu 42 Nov 30, 2022
This is an early in-development version of training CLIP models with hivemind.

A transformer that does not hog your GPU memory This is an early in-development codebase: if you want a stable and documented hivemind codebase, look

<a href=[email protected]"> 4 Nov 06, 2022
Compares various time-series feature sets on computational performance, within-set structure, and between-set relationships.

feature-set-comp Compares various time-series feature sets on computational performance, within-set structure, and between-set relationships. Reposito

Trent Henderson 7 May 25, 2022
ECCV18 Workshops - Enhanced SRGAN. Champion PIRM Challenge on Perceptual Super-Resolution. The training codes are in BasicSR.

ESRGAN (Enhanced SRGAN) [ ๐Ÿš€ BasicSR] [Real-ESRGAN] โœจ New Updates. We have extended ESRGAN to Real-ESRGAN, which is a more practical algorithm for rea

Xintao 4.7k Jan 02, 2023
Reference implementation for Structured Prediction with Deep Value Networks

Deep Value Network (DVN) This code is a python reference implementation of DVNs introduced in Deep Value Networks Learn to Evaluate and Iteratively Re

Michael Gygli 55 Feb 02, 2022
Learning Temporal Consistency for Low Light Video Enhancement from Single Images (CVPR2021)

StableLLVE This is a Pytorch implementation of "Learning Temporal Consistency for Low Light Video Enhancement from Single Images" in CVPR 2021, by Fan

99 Dec 19, 2022
PyTorch source code for Distilling Knowledge by Mimicking Features

LSHFM.detection This is the PyTorch source code for Distilling Knowledge by Mimicking Features. And this project contains code for object detection wi

Guo-Hua Wang 4 Dec 17, 2022
PyTorch implementation of Higher Order Recurrent Space-Time Transformer

Higher Order Recurrent Space-Time Transformer (HORST) This is the official PyTorch implementation of Higher Order Recurrent Space-Time Transformer. Th

13 Oct 18, 2022
PyTorch implementation for Graph Contrastive Learning with Augmentations

Graph Contrastive Learning with Augmentations PyTorch implementation for Graph Contrastive Learning with Augmentations [poster] [appendix] Yuning You*

Shen Lab at Texas A&M University 382 Dec 15, 2022
Revisiting Global Statistics Aggregation for Improving Image Restoration

Revisiting Global Statistics Aggregation for Improving Image Restoration Xiaojie Chu, Liangyu Chen, Chengpeng Chen, Xin Lu Paper: https://arxiv.org/pd

MEGVII Research 128 Dec 24, 2022
Code for the submitted paper Surrogate-based cross-correlation for particle image velocimetry

Surrogate-based cross-correlation (SBCC) This repository contains code for the submitted paper Surrogate-based cross-correlation for particle image ve

5 Jun 30, 2022
A compendium of useful, interesting, inspirational usage of pandas functions, each example will be an ipynb file

Pandas_by_examples A compendium of useful/interesting/inspirational usage of pandas functions, each example will be an ipynb file What is this reposit

Guangyuan(Frank) Li 32 Nov 20, 2022
All supplementary material used by me while TA-ing CS3244: Machine Learning

CS3244-Tutorial-Material All supplementary material used by me while TA-ing CS3244: Machine Learning at NUS School of Computing. What is this? I teach

Rishabh Anand 18 Sep 23, 2022
VolumeGAN - 3D-aware Image Synthesis via Learning Structural and Textural Representations

VolumeGAN - 3D-aware Image Synthesis via Learning Structural and Textural Representations 3D-aware Image Synthesis via Learning Structural and Textura

GenForce: May Generative Force Be with You 116 Dec 26, 2022
classify fashion-mnist dataset with pytorch

Fashion-Mnist Classifier with PyTorch Inference 1- clone this repository: git clone https://github.com/Jhamed7/Fashion-Mnist-Classifier.git 2- Instal

1 Jan 14, 2022
Count GitHub Stars โญ

Count GitHub Stars per Day โญ Track GitHub stars per day over a date range to measure the open-source popularity of different repositories. Requirement

Ultralytics 20 Nov 20, 2022