wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity
1,737 Commits 7 Branches 68 Tags
tb
Commit Graph
1737 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
William Falcon d794ee4522 Merge branch 'master' into tb 2020-01-13 22:13:59 -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
Ayla Khan 1969c6cc2a Remove extraneous f character from f-string. (#679) 
Makes tracking experiment names confusion, especially when using uuids.
 2020-01-13 22:11:04 -05:00
Jirka Borovec db6b404748 CI pass (#671) 
* fix pillow in test

* test acc

* update version in deprecated msg
 2020-01-13 22:09:47 -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
Verena Haunschmid 9ac91adea9 Update requirements.txt (#664) 
Fix typo 'buildins' -> 'builtins'
 2020-01-05 14:34:44 -05:00
Nic Eggert 019f612204 Fix amp tests (#661) 
* Run AMP tests in their own process

With opt_level="O1" (the default), AMP patches many
torch functions, which breaks any tests that run afterwards.
This patch introduces a pytest extension that lets
tests be marked with @pytest.mark.spawn so that they
are run in their own process using torch.multiprocessing.spawn
so that the main python interpreter stays un-patched.

Note that tests using DDP already run AMP in its own process,
so they don't need this annotation.

* Fix AMP tests

Since AMP defaults to O1 now, DP tests no longer throw exceptions.

Since AMP patches torch functions, CPU inference no longer works.
Skip prediction step for AMP tests.

* typo
 2020-01-05 14:34:25 -05:00
Jirka Borovec c32f2b9116 Fix ci xos (#647) 
* upgrade python 3.7

* upgrade python 3.7
 2019-12-22 21:18:09 -05:00
Hao Sheng ca73b70d15 fix of issue 600 (#625) 2019-12-14 20:24:46 -08:00
Jeremy Jordan 3dd0b8c186 fix metric name to work with default earlystopping (#628) 2019-12-14 20:23:44 -08:00
William Falcon 8c5d66196b Update README.md 2019-12-12 11:08:02 -08:00
William Falcon d44c91d854 Update README.md 2019-12-12 11:07:06 -08:00
William Falcon af6d552d35 Update README.md 2019-12-12 11:06:20 -08:00
William Falcon 24bfa53894 Update README.md 2019-12-12 13:36:17 -05:00
William Falcon 52295986e4 Update README.md 2019-12-12 13:35:41 -05:00
William Falcon 64c428ec49 Update README.md 2019-12-12 13:33:49 -05:00
William Falcon a6fc172387 Update README.md 2019-12-12 10:24:53 -08:00
William Falcon be43fbb918 Update README.md 2019-12-12 10:23:49 -08:00
William Falcon 15cb79923a Add files via upload 2019-12-12 10:23:02 -08:00
Jay Morgan d1633aac11 Fix #618 Change papi to api (#619) 
* Change papi to api

* Added try catch for old/new api reference
 2019-12-10 16:24:21 -08: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
VSJMilewski d562172b4c Allow for multiple example inputs when creating summary (#543) 2019-12-09 04:42:07 -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 607dbdaefd update GitHub templates (#612) 2019-12-08 17:07:24 -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
William Falcon 4c7cfd3f12 Update README.md 2019-12-08 00:09:16 -08:00
William Falcon 131503a15a Update README.md 2019-12-08 00:09:03 -08:00
William Falcon 99c9b82527 made tensorboard the default not test-tube 2019-12-07 23:36:51 -05:00
William Falcon 47a82cf1b9 refactor 2019-12-07 23:32:17 -05:00
William Falcon 1b86ed9cc3 refactor 2019-12-07 23:31:47 -05:00
William Falcon 94bd2ae3e1 refactor 2019-12-07 23:30:59 -05:00
Nic Eggert 5329c72cb0 Implement TensorboardLogger (#607) 
* Implement TensorboardLogger

* Pass default_save_path to trainers

* Update tensorboard.py
 2019-12-07 23:25:37 -05:00
Nic Eggert 2baa80d626 Make sure train doesn't crash when called at max_epoch (#608) 2019-12-07 23:22:03 -05:00
Jirka Borovec 4970624f8b fix Logger tests for Win (#605) 
* fix mlflow test

* fix mlflow test

* update logger / mlflow

* flake8

* fix appveyor
 2019-12-07 19:25:12 -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
Jirka Borovec c6e0dbedd0 prevent Travis caching (#590) 
* change CI install

* change CI install

* change CI install
 2019-12-07 08:49:10 -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
Nic Eggert 0489e31b02 Fix CometML tests (#585) 
* monkeypatch atexit.register to fix problem with cometml logging

* Use experiment id for version in cometml
 2019-12-07 00:24:59 -05:00
Jirka Borovec c374c4fb80 extend documentation (#569) 
* extend documentation

* update index

* fix list
 2019-12-07 00:23:48 -05:00
Jirka Borovec ed97231e09 update GitHub templates (#601) 2019-12-07 00:23:01 -05:00
Jirka Borovec 6666ca5af3 add slack badge (#583) 
* add slack badge

* Update README.md
 2019-12-04 19:21:52 -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
William Falcon 6ba30a113d fixed gan template (#528) 
* fixed gan template

* Update gan.py
 2019-12-04 08:28:46 -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