Skip to content
/ GOS Public

Grassroots Operator Search Implementation

Notifications You must be signed in to change notification settings

IHIaadj/GOS

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GOS: Grassroot Operator Search

Grassroot Operator Search (GOS) is a PyTorch project that aims to find the optimal operator replacement for the least efficient layer in a deep learning architecture. By performing an operator search at the grassroots level, GOS seeks to enhance the overall efficiency and performance of deep learning models.

Overview

Deep learning models often consist of multiple layers, each performing different operations such as convolution, pooling, or fully connected layers. However, some layers may be less efficient in terms of computation or memory usage, leading to suboptimal performance. GOS addresses this issue by focusing on identifying and replacing the least efficient layer with an alternative operator that improves efficiency without sacrificing accuracy.

GOS utilizes a search algorithm to explore various operator replacements and evaluates their impact on model performance and efficiency metrics. The project provides a flexible framework to conduct operator search experiments on different deep learning architectures and datasets.

https://github.com/IHIaadj/GOS/blob/master/overview.png

Key Features

  • Automatic identification of the least efficient layer in a deep learning architecture.
  • Operator search to find the optimal replacement for the identified layer.
  • Evaluation of operator replacements based on performance and efficiency metrics.
  • Flexible and extensible framework for conducting experiments on various architectures and datasets.
  • Easy integration with PyTorch models.

Getting Started

Prerequisites

  • Python 3.7
  • PyTorch

Installation

git clone https:// anonymous
pip install -r requirements.txt

Usage

  • Prepare your deep learning model and dataset according to the project structure.
  • Configure the GOS settings in the config.py file, including the search algorithm, performance metrics, and other parameters.
  • Run the GOS script:
python gos.py
  • GOS will perform the operator search, evaluate different replacements, and provide results and recommendations.

Contributions

Contributions to the GOS project are welcome! If you encounter any issues or have suggestions for improvements, please open an issue or submit a pull request. Let's collaborate to enhance the efficiency and performance of deep learning models through operator search.

LICENSE

Feel free to customize the content based on your specific project details, including the prerequisites, installation instructions, usage guidelines, and contribution guidelines.

About

Grassroots Operator Search Implementation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published