Commit Graph
142 Commits
Author SHA1 Message Date
William Falcon 57db191028 flake 8 2020-01-17 05:03:31 -05:00
William Falcon 36da61eb01 fixed lightning import 2020-01-16 14:51:54 -05:00
William Falcon f353b021b5 working on trainer docs 2020-01-16 10:51:15 -05:00
William Falcon d05f805b25 working on trainer docs 2020-01-16 10:47:39 -05:00
William Falcon 1c7a220432 working on trainer docs 2020-01-16 10:47:12 -05:00
William Falcon ed91ad6407 working on trainer docs 2020-01-16 10:41:05 -05:00
William Falcon 62d213ab3c working on trainer docs 2020-01-16 10:40:16 -05:00
William Falcon 19e13e776b working on trainer docs 2020-01-16 10:33:19 -05:00
William Falcon 59e22a9225 working on trainer docs 2020-01-16 10:32:41 -05:00
William Falcon cebcd3039c set auto dp if no backend 2020-01-16 10:25:51 -05:00
William Falcon 0b0f2c01e1 working on trainer docs 2020-01-16 10:16:17 -05:00
William Falcon 1b12cb83dc working on trainer docs 2020-01-16 10:14:29 -05:00
William Falcon 80454d3635 working on trainer docs 2020-01-16 10:13:48 -05:00
William Falcon 8883f031ac working on trainer docs 2020-01-16 09:55:55 -05:00
William Falcon 87d9c21eb7 working on trainer docs 2020-01-16 09:50:13 -05:00
William Falcon 365558c824 working on trainer docs 2020-01-16 09:26:35 -05:00
William Falcon f700912f72 making private members 2020-01-16 08:26:53 -05:00
William Falcon f80d24c188 making private members 2020-01-16 08:11:04 -05:00
William Falcon 145ea2713d making private members 2020-01-16 08:06:12 -05:00
William Falcon d247431300 finished rebase 2020-01-16 07:45:36 -05:00
William Falcon 3e5c5f9588 added direct links to docs 2020-01-16 07:27:31 -05:00
William Falcon 637f2344de added direct links to docs 2020-01-16 07:27:31 -05:00
William Falcon 31a3854e6a added direct links to docs 2020-01-16 07:27:22 -05:00
William Falcon f9285787a1 fixed left menu 2020-01-16 07:27:02 -05:00
William Falcon c6c67a34ad finished callbacks 2020-01-16 07:27:02 -05:00
William Falcon 24897506a0 finished callbacks 2020-01-16 07:26:45 -05:00
William Falcon 0b416e96b0 updated docs 2020-01-16 07:25:31 -05:00
Jirka Borovec f72e354ee6 fixing TensorBoard (#687)
* flake8

* fix typo

* fix tensorboardlogger
drop test_tube dependence

* formatting

* fix tensorboard & tests

* upgrade Tensorboard

* test formatting separately

* try to fix JIT issue

* add tests for 1.4
2020-01-16 07:22:29 -05:00
William Falcon 88b750a018 default logger is now tensorboard (#609)
* refactor

* refactor

* refactor

* made tensorboard the default not test-tube
2020-01-14 14:40:41 -05:00
Vadim Bereznyuk 756c70a4a0 Clearer disable validation logic (#650)
* Clearer disable validation logic

* fix for fast_dev_run

* flake8 fix

* Test check fix

* update error message
2020-01-13 22:31:15 -05:00
Jirka Borovec f7db44e750 fix deprecated tng and abstract ligntning (#644) 2020-01-13 22:20:38 -05:00
Ayberk Aydın 0ae3dd9ed4 Fix GAN training. (#603)
* fix dangling gradients

make sure only the gradients of the current optimizer's paramaters are calculated in the training step.

* add note about multiple optimizer gradient update

* Update training_loop.py
2020-01-13 22:12:04 -05:00
Vadim Bereznyuk 12edc3099c Fix the number of training batches used in the training loop (#653)
* Fix the number of processed training batches

* Fix tests

* fix tests

* fix tests

* One more attempt

* Fix another test
2020-01-05 14:37:09 -05:00
Vadim Bereznyuk 7824b5c5f5 Fix percent_checks (#649)
* fix percent_checks

* Added _percent_range_check

* remove max
2020-01-05 14:36:06 -05:00
Adrian Wälchli e2ee4ddbdb Fix early stopping off by 2 (min_epochs) (#617)
* fix early stopping off by 2

* add min_epochs example in docs
2019-12-09 10:32:49 -08:00
Elliot Waite b492e2b89e Change nb to num in ABCs, comments, and tqdm logging (#613)
* Change nb to num in ABCs, comments, and tqdm logging

* Fix warnings text

* Make warnings one line

* Change num to number in comments
2019-12-09 04:40:26 -08:00
Jirka Borovec 5d00e62047 Fix logger, tensorboard (#610)
* fix logger tests

* fix missing flush

* fix tensorboard

* fix namespace

* fix flush

* fix add_hparams
2019-12-08 07:59:25 -08:00
Nic Eggert 2baa80d626 Make sure train doesn't crash when called at max_epoch (#608) 2019-12-07 23:22:03 -05:00
ctlaltdefeat 58cc6e13b9 Update logging.py (#602) 2019-12-07 10:12:33 -05:00
schwobr 2f01c03b38 Additional hooks (#598)
* Renamed `on_sanity_check_start` to `on_train_start` and added `on_train_end` to `ModelHooks`

* changed tests to use `on_train_start` instead of `on_sanity_check_start`
2019-12-07 08:52:06 -05:00
Elliot Waite 1051c189e1 Simplify variables: step, epoch, max_epochs, min_epochs (#589) 2019-12-07 08:50:21 -05:00
Adrian Wälchli f7e1040236 Docs and Tests for "gpus" Trainer Argument (#593)
* add table for gpus argument

* fix typo in error message

* tests for supported values

* tests for unsupported values

* fix typo

* add table for gpus argument

* fix typo in error message

* tests for supported values

* tests for unsupported values

* fix typo

* fix typo list->str

* fix travis warning "line too long"
2019-12-07 08:48:45 -05:00
YehCF cc65f39d97 Fix number of total steps shown in progress bar during sanity validation check when number of validation dataloaders >= 2 (#597)
* type: debug

Calculate the adequate number of steps to run during sanity_check.
This fixes the bug when there are two or more validation dataloaders.

- Before: total=self.num_sanity_val_steps
- After: total=self.num_sanity_val_steps*len(self.get_val_dataloaders())

* type: refactor

Put total=... in the next line

* type: refactor

run flake8
2019-12-07 08:47:59 -05:00
Jirka Borovec 1d4b6be17b rename trainer modules, drop _mixin (#571)
* rename trainer modules, drop _mixin

* fix imports
2019-12-04 11:39:14 -05:00
Jirka Borovec e0dbc8ab46 Abstract Mixin classes (#572)
* make partial Trainer classes as abstract

* add empty attributes/methods

* flake8

* fix mixin order

* update abstact

* reorder
2019-12-04 10:57:32 -05:00
Adrian Wälchli 218f0a5b4a inspect training_step for opt_idx (#573) 2019-12-04 07:32:47 -05:00
Ir1dXD c316173e89 use print for INFO and lower levels summarize() (#580)
* use print for INFO and lower levels summarize()

* use logging.INFO instead of magic number

* bring logging.info back for other cases

* move logging config to __init__.py

* prepend the model summary with a newline
2019-12-04 07:05:34 -05:00
Ir1dXD d4571d1d6f filter param with no grad (#579) 2019-12-04 07:04:58 -05:00
Jirka Borovec ab4fea0b55 fix defecation warnings (#570)
* fix defecation warnings

* flake8

* update deprecations
2019-12-04 06:59:19 -05:00
Jirka Borovec 3a58937d8b rename variables nb -> num (#567)
* rename nb -> num

* flake8

* batch_nb, epoch_nb, gpu_nb, split_nb

* add _num deprecations
2019-12-04 06:57:10 -05:00