Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Feat: Extend ptq_evaluate to allow switching between torchvision and timm datsets #946

Draft
wants to merge 14 commits into
base: dev
Choose a base branch
from
Draft

Feat: Extend ptq_evaluate to allow switching between torchvision and timm datsets #946

wants to merge 14 commits into from

Commits on Apr 30, 2024

  1. modified ptq_evaluate.py to allow switching between timm and torchvis… 

    …ion models
    @costigt-dev
    costigt-dev committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    11cb2ed View commit details
    Browse the repository at this point in the history

  2. Merge branch 'dev' of github.com:Xilinx/brevitas into ptq_timm

    @costigt-dev
    costigt-dev committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    2acf6be View commit details
    Browse the repository at this point in the history

  3. updated node.kwargs to return false if not present so ptq_evaluate wo… 

    …rks with more timm models
    @costigt-dev
    costigt-dev committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    863a4e5 View commit details
    Browse the repository at this point in the history

  4. implemented mean in int_quant_tensor

    @costigt-dev
    costigt-dev committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    c6b5cd5 View commit details
    Browse the repository at this point in the history

  5. added variable number of arguments to mean implementation

    @costigt-dev
    costigt-dev committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    5469a24 View commit details
    Browse the repository at this point in the history

  6. args and kwargs added to mean

    @costigt-dev
    costigt-dev committed Apr 30, 2024
    Configuration menu
    Copy the full SHA
    5596ca4 View commit details
    Browse the repository at this point in the history

Commits on May 1, 2024

  1. using timm model config info to load data and renamed the dataset par… 

    …ameter to repository
    @costigt-dev
    costigt-dev committed May 1, 2024
    Configuration menu
    Copy the full SHA
    e29ef29 View commit details
    Browse the repository at this point in the history

  2. modified dataloader code to use timm data_config

    @costigt-dev
    costigt-dev committed May 1, 2024
    Configuration menu
    Copy the full SHA
    f5d5e8e View commit details
    Browse the repository at this point in the history

  3. fix typo

    @costigt-dev
    costigt-dev committed May 1, 2024
    Configuration menu
    Copy the full SHA
    c0f596b View commit details
    Browse the repository at this point in the history

  4. moved pre quantize validate step to earlier in the script

    @costigt-dev
    costigt-dev committed May 1, 2024
    Configuration menu
    Copy the full SHA
    28e8868 View commit details
    Browse the repository at this point in the history

  5. moved offloading model to gpu to earlier

    @costigt-dev
    costigt-dev committed May 1, 2024
    Configuration menu
    Copy the full SHA
    5ae5719 View commit details
    Browse the repository at this point in the history

  6. implement squeeze in int_quant_tensor

    @costigt-dev
    costigt-dev committed May 1, 2024
    Configuration menu
    Copy the full SHA
    8283979 View commit details
    Browse the repository at this point in the history

Commits on May 3, 2024

  1. added sigmoid function to QuantTensor

    @costigt-dev
    costigt-dev committed May 3, 2024
    Configuration menu
    Copy the full SHA
    6390416 View commit details
    Browse the repository at this point in the history

Commits on May 9, 2024

  1. added sum function to int_quant_tensor

    @costigt-dev
    costigt-dev committed May 9, 2024
    Configuration menu
    Copy the full SHA
    2ca8133 View commit details
    Browse the repository at this point in the history