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:
Jirka Borovec
2020-04-09 14:05:46 -04:00
committed by GitHub
parent b4eb3884cf
commit 17f58d2e11
41 changed files with 213 additions and 187 deletions
+1 -2
View File
@@ -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,