wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

CI pass (#671)

Browse Source 
* fix pillow in test

* test acc

* update version in deprecated msg
This commit is contained in:
Jirka Borovec
2020-01-13 22:09:47 -05:00
committed by William Falcon
parent 12edc3099c
commit db6b404748
12 changed files with 23 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/requirements.txt
+2 -1
View File
@@ -8,4 +8,5 @@ check-manifest
# test_tube # already installed in main req.
mlflow
comet_ml
twine==1.13.0
twine==1.13.0
pillow<7.0.0
tests/test_cpu_models.py
+1 -1
View File
@@ -97,7 +97,7 @@ def test_running_test_after_fitting(tmpdir):
trainer_options = dict(
default_save_path=tmpdir,
show_progress_bar=False,
max_epochs=1,
max_epochs=4,
train_percent_check=0.4,
val_percent_check=0.2,
test_percent_check=0.2,
tests/test_restore_models.py
+3 -3
View File
@@ -153,7 +153,7 @@ def test_running_test_pretrained_model_dp(tmpdir):
trainer_options = dict(
show_progress_bar=True,
max_epochs=1,
max_epochs=4,
train_percent_check=0.4,
val_percent_check=0.2,
checkpoint_callback=checkpoint,
@@ -269,7 +269,7 @@ def test_cpu_restore_training(tmpdir):
logger = tutils.get_test_tube_logger(tmpdir, False, version=test_logger_version)
trainer_options = dict(
max_epochs=4,
max_epochs=8,
val_check_interval=0.50,
val_percent_check=0.2,
train_percent_check=0.2,
@@ -290,7 +290,7 @@ def test_cpu_restore_training(tmpdir):
# we want to see if the weights come back correctly
new_logger = tutils.get_test_tube_logger(tmpdir, False, version=test_logger_version)
trainer_options = dict(
max_epochs=4,
max_epochs=2,
val_check_interval=0.50,
val_percent_check=0.2,
train_percent_check=0.2,