Skip to content

Commit 06d120f

Browse files
committed
Update README.md
1 parent da190e8 commit 06d120f

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

README.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,10 @@
44

55
I got data from [here](https://drive.google.com/file/d/0BxDIywue_VABY1dRcFVvZ3BodnM/view?usp=sharing). Download it and name it as `data`.
66

7+
## Model
8+
9+
You can download a pretrained model from [here](https://github.com/tae898/room-classification/releases/download/v0.1/model.ckpt). It's a fine-tuned [EfficientNet-b3](https://github.com/lukemelas/EfficientNet-PyTorch) using [pytorch-lightning](https://www.pytorchlightning.ai/)
10+
711
## Prerequisites
812

913
1. A unix or unix-like x86 machine

0 commit comments

Comments
 (0)