Paper_Review
├── GhostNet_pytorch
├── datasets
│ └── rsp_data
├── data
│ ├── __init__.py
│ └── dataset.py
├── option.py
├── model.py
├── hubconf.py
├── train.py
├── inference.py
└── README.md
data/__init__.py
: dataset loader
data/dataset.py
: data preprocess & get item
model.py
: Define block and construct Model
option.py
: Environment setting
hubconf.py
: model hub
Clone repo and install depenency
# 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)
python3 inference.py
--device {}(defautl: cpu) \
--data_name {}(default: rsp_data) \
--num_workers {}(default: 4) \
--eval_batch_size {}(default: 16)