mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
@@ -1,6 +1,6 @@
|
||||
name: CI testing
|
||||
|
||||
# https://help.github.com/en/actions/reference/events-that-trigger-workflows
|
||||
# see: https://help.github.com/en/actions/reference/events-that-trigger-workflows
|
||||
on:
|
||||
# Trigger the workflow on push or pull request,
|
||||
# but only for the master branch
|
||||
@@ -116,6 +116,7 @@ jobs:
|
||||
python setup.py check --metadata --strict
|
||||
python setup.py sdist
|
||||
twine check dist/*
|
||||
|
||||
#- name: Try install package
|
||||
# if: ! startsWith(matrix.os, 'windows')
|
||||
# run: |
|
||||
|
||||
@@ -14,6 +14,8 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
|
||||
|
||||
### Removed
|
||||
|
||||
- Removed Warning from trainer loop ([#1634](https://github.com/PyTorchLightning/pytorch-lightning/pull/1634))
|
||||
|
||||
### Fixed
|
||||
|
||||
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
"""Root package info."""
|
||||
|
||||
__version__ = '0.7.4'
|
||||
__version__ = '0.7.5rc1'
|
||||
__author__ = 'William Falcon et al.'
|
||||
__author_email__ = 'waf2107@columbia.edu'
|
||||
__license__ = 'Apache-2.0'
|
||||
|
||||
@@ -299,9 +299,6 @@ class TrainerTrainLoopMixin(ABC):
|
||||
"""Warning: this is just empty shell for code implemented in other class."""
|
||||
|
||||
def train(self):
|
||||
rank_zero_warn('Displayed epoch numbers in the progress bar start from "1" until v0.6.x,'
|
||||
' but will start from "0" in v0.8.0.', RuntimeWarning)
|
||||
|
||||
# get model
|
||||
model = self.get_model()
|
||||
|
||||
|
||||
@@ -120,12 +120,13 @@ def test_no_val_end_module(tmpdir):
|
||||
new_weights_path = os.path.join(tmpdir, 'save_test.ckpt')
|
||||
trainer.save_checkpoint(new_weights_path)
|
||||
|
||||
# load new model
|
||||
tags_path = tutils.get_data_path(logger, path_dir=tmpdir)
|
||||
tags_path = os.path.join(tags_path, 'meta_tags.csv')
|
||||
# assert ckpt has hparams
|
||||
ckpt = torch.load(new_weights_path)
|
||||
assert 'hparams' in ckpt.keys(), 'hparams missing from checkpoints'
|
||||
|
||||
# won't load without hparams in the ckpt
|
||||
model_2 = LightningTestModel.load_from_checkpoint(
|
||||
checkpoint_path=new_weights_path,
|
||||
tags_csv=tags_path
|
||||
)
|
||||
model_2.eval()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user