wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-26 13:00:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed basic trainer

Browse Source
This commit is contained in:
William Falcon committed 2019-06-25 18:11:13 -04:00
1 parent 6c705a0525
commit 983551653d
1 file changed
+4 -4
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/examples/basic_trainer.py
+4 -4
View File
@@ -2,10 +2,10 @@ import os
import sys
from test_tube import HyperOptArgumentParser, Experiment
from pytorch-lightning.models.trainer import Trainer
from pytorch-lightning.utils.arg_parse import add_default_args
from pytorch-lightning.utils.pt_callbacks import EarlyStopping, ModelCheckpoint
from demo.example_model import ExampleModel
from pytorch_lightning.models.trainer import Trainer
from pytorch_lightning.utils.arg_parse import add_default_args
from pytorch_lightning.utils.pt_callbacks import EarlyStopping, ModelCheckpoint
from docs.source.examples.example_model import ExampleModel
def main(hparams):