-
Notifications
You must be signed in to change notification settings - Fork 27.3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[lightning_base] fix s2s logging, only make train_loader once #6404
Conversation
@@ -31,3 +34,24 @@ def test_run_ner(self): | |||
with patch.object(sys, "argv", ["run.py"] + testargs): | |||
result = run_ner.main() | |||
self.assertLess(result["eval_loss"], 1.5) | |||
|
|||
|
|||
def test_run_ner_pl(self): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this doesn't test lightning and should be in a separate PR.
@@ -161,13 +161,6 @@ def add_model_specific_args(parser, root_dir): | |||
type=int, | |||
help="The number of GPUs allocated for this, it is by default 0 meaning none", | |||
) | |||
parser.add_argument( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
centralize data_dir argument to add_generic_args
cause rule of 3
Codecov Report
@@ Coverage Diff @@
## master #6404 +/- ##
==========================================
- Coverage 80.38% 80.26% -0.12%
==========================================
Files 156 156
Lines 28058 28058
==========================================
- Hits 22554 22521 -33
- Misses 5504 5537 +33
Continue to review full report at Codecov.
|
…huggingface#6404)" This reverts commit 59da5d3.
setup is called many times (incl twice by trainer.test), creating a dataloader each time. Will only creating a train_loader on the first call cause bad side effects that I don't understand @nateraw @williamFalcon ?
I read docs [https://pytorch-lightning.readthedocs.io/en/latest/introduction_guide.html], so I think I'm fine, but not sure.
cc @stas00
Also:
data_dir
argument toadd_generic_args
cause rule of 3Checks: