Spaces:
Runtime error
Runtime error
1. Understand the file structure: | |
1. architectures in `discriminator.py` `generator.py` `classifier.py` | |
2. data loading in `data.py` | |
3. data transformation `transforms.py` | |
4. optimizers in `optim.py` | |
5. utilities in `utils.py` | |
6. training procedure in `trainer.py` | |
2. Write **tests** in `tests/` | |
1. your file should match `test_*.py` | |
2. update existing tests when adding functionalities | |
3. run tests regularly to check you haven't broken anything `python tests/run.py` | |
3. Add **WIP** in your PR's title when not ready to merge | |
5. Open an Issue if something's odd, or to assign yourself a todo | |
6. **Format your code** with [black](https://github.com/psf/black) | |
7. Only update `trainer/defaults.yaml` with values that should be shared across runs and users | |
1. use `config/trainer/local_tests.yaml` or any other to setup your particular config overriding `trainer/defaults.yaml` | |
## Running tests | |
As per `7.` you should set your particular config in `config/local_tests.yaml`. Mine looks like: | |
```yaml | |
output_path: /Users/victor/Documents/ccai/github/climategan/example_data | |
# ------------------- | |
# ----- Tasks ----- | |
# ------------------- | |
#tasks: [a, d, h, s, t, w] | |
tasks: [a, d, s, t] # for now no h or w | |
# ---------------- | |
# ----- Data ----- | |
# ---------------- | |
data: | |
files: # if one is not none it will override the dirs location | |
base: /Users/victor/Documents/ccai/github/climategan/example_data | |
transforms: | |
- name: hflip | |
ignore: false | |
p: 0.5 | |
- name: resize | |
ignore: false | |
new_size: 256 | |
- name: crop | |
ignore: false | |
height: 64 | |
width: 64 | |
gen: | |
encoder: | |
n_res: 1 | |
default: | |
n_res: 1 | |
train: | |
log_level: 1 | |
``` | |
Setting `n_res` to 1 is important to run tests faster and with less memory | |