mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-11 12:31:23 +08:00
add rank warning (#1428)
* add rank warning * changelog * use rank_zero_warn * user trainer_init * replace warnings * fix test * flake8 * docs * changelog * bug lol
This commit is contained in:
+1
-2
@@ -1,5 +1,4 @@
|
||||
import os
|
||||
import warnings
|
||||
from argparse import Namespace
|
||||
|
||||
import numpy as np
|
||||
@@ -92,7 +91,7 @@ def run_model_test(trainer_options, model, on_gpu=True):
|
||||
|
||||
|
||||
def get_default_hparams(continue_training=False, hpc_exp_number=0):
|
||||
tests_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
|
||||
_ = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
|
||||
|
||||
args = {
|
||||
'drop_prob': 0.2,
|
||||
|
||||
Reference in New Issue
Block a user