-
-
Notifications
You must be signed in to change notification settings - Fork 24
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
refactor: Reorganized references folder (#104)
* 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
1 parent
217d461
commit aab603e
Showing
10 changed files
with
58 additions
and
518 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -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 | ||
|
This file was deleted.
Oops, something went wrong.
This file was deleted.
Oops, something went wrong.
Oops, something went wrong.