Skip to content

Commit

Permalink
refactor: Reorganized references folder (#104)
Browse files Browse the repository at this point in the history
* reorganize references folder

* allow training on openfire

* remove utils in docs

* update use open fire parameter

* missing change with new use_openfire

* use _ instead of -

* remove useless download = False

* remove useless if

* update readme

* already change in another PR
  • Loading branch information
MateoLostanlen authored Mar 1, 2021
1 parent 217d461 commit aab603e
Show file tree
Hide file tree
Showing 10 changed files with 58 additions and 518 deletions.
23 changes: 17 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -86,18 +86,29 @@ docker build . -t <YOUR_IMAGE_TAG>
You are free to use any training script, but some are already provided for reference. In order to use them, install the specific requirements and check script options as follows:

```shell
pip install -r references/classification/OpenFire/fastai/requirements.txt
python references/classification/OpenFire/fastai/train.py --help
pip install -r references/requirements.txt
python references/classification/train.py --help
```

You can then run the script with your own arguments:
You can then use the script to train tour model on one of our datasets:

### Wildfire

Download Dataset from https://drive.google.com/file/d/1Y5IyBLA5xDMS1rBdVs-hsVNGQF3djaR1/view?usp=sharing

This dataset is protected by a password, please contact us at [email protected]

```shell
python references/classification/OpenFire/fastai/train.py --lr 3e-3 --epochs 4 --pretrained --deterministic
```
python train.py WildFireLght/ --model rexnet1_0x --lr 1e-3 -b 16 --epochs 20 --opt radam --sched onecycle --device 0
```

### OpenFire

*Please note that most tasks are provided with two training scripts (and their `requirements.txt`): one using [fastai](https://github.com/fastai/fastai) and the other without it.*
You can also use out opensource dataset without password

```
python train.py OpenFire/ --use-openfire --model rexnet1_0x --lr 1e-3 -b 16 --epochs 20 --opt radam --sched onecycle --device 0
```


## Documentation
Expand Down
3 changes: 0 additions & 3 deletions references/classification/OpenFire/fastai/requirements.txt

This file was deleted.

179 changes: 0 additions & 179 deletions references/classification/OpenFire/fastai/train.py

This file was deleted.

Loading

0 comments on commit aab603e

Please sign in to comment.