Domain-exploration

Exploring the role of domain in training and performance of DL models.

Installation

Clone the repo

git clone [email protected]:reghbali/domain-exploration.git
cd domain-exploration

Then run (we recommend using a python virtual environment)

pip install --upgrade pip

Update the setuptools

pip install --upgrade setuptools

Run the following command to install the domain-exploration library:

pip install -e .

Training

Run

python main.py fit [--data DATA_MODULE] [--data.domain DOMAIN] [--model MODEL]

for instance:

python main.py fit --data MNISTDataModule --data.domain pixel --model MLP_MNIST

and for freq domain training:

python main.py fit --data MNISTDataModule --data.domain freq --model MLP_MNIST