├── GhostNet_pytorch
        ├── datasets
        │     └── rsp_data
        ├── data
        │     ├── __init__.py
        │     └── dataset.py
        ├── option.py
        ├── model.py
        ├── hubconf.py
        ├── train.py
        ├── inference.py
        └── README.md

# Clone this repo and install dependency
git clone https://github.com/inhopp/GhostNet_pytorch.git
pip3 install requirements.txt

python3 train.py
    --device {}(defautl: cpu) \
    --data_name {}(default: rsp_data) \
    --lr {}(default: 0.0005) \
    --n_epoch {}(default: 10) \
    --num_workers {}(default: 4) \
    --batch_size {}(default: 32) \ 
    --eval_batch_size {}(default: 16)