Skip to content

akkshita/Chargrid-pytorch

Repository files navigation

Installation

After cloning this repository

Use the package manager pip to install the requirements.

pip install -r requirements.txt

Copy env.sample to .env Then add folder paths for the dataset in .env file

Usage

Run following to preprocess the dataset

python -m data_pipeline.preprocessing
python -m data_pipeline.preprocessing_bis
python -m data_pipeline.preprocessing_ter

Once you have data preprocessed run train.py.

python train.py

Or for resumable training run

python resumable.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published