Skip to content

Commit 1a75518

Browse files
add fine tuning
1 parent c584dc9 commit 1a75518

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

.main.py.swp

16 KB
Binary file not shown.

README.md

+7-2
Original file line numberDiff line numberDiff line change
@@ -26,15 +26,20 @@ DenseNet201 | 95.50%
2626

2727
## Getting started
2828

29-
**need GPUS**
30-
29+
### Train from scratch
3130

3231
```bash
3332
python main.py --help
3433
python main.py --models=densenet121 --gpu=0 --visible=0,1,2
3534
```
3635
model will be saved in checkpoints/
3736

37+
### Fine-tuning
38+
39+
```bash
40+
python main.py --models=densenet121 --gpu=0 --visible=0,1,2 --resume
41+
```
42+
3843
## Pretrained Models
3944

4045
CNN Architecture | Model

0 commit comments

Comments
 (0)