Uncomment plot_lr_scheduler in train() and pass log_dir as save location |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| activations.py | ||
| datasets.py | ||
| google_utils.py | ||
| torch_utils.py | ||
| utils.py | ||
Uncomment plot_lr_scheduler in train() and pass log_dir as save location |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| activations.py | ||
| datasets.py | ||
| google_utils.py | ||
| torch_utils.py | ||
| utils.py | ||