Browse Source

set random seed

tags/v0.4.10
QipengGuo 5 years ago
parent
commit
205f8fe6af
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      reproduction/Star_transformer/train.py

+ 2
- 2
reproduction/Star_transformer/train.py View File

@@ -109,8 +109,8 @@ class MyCallback(FN.core.callback.Callback):


def train():
#seed = set_rng_seeds(1234)
seed = set_rng_seeds(np.random.randint(65536))
seed = set_rng_seeds(28848)
#seed = set_rng_seeds(np.random.randint(65536))
print('RNG SEED {}'.format(seed))
print('loading data')
ds_list, word_v, tag_v = g_datasets['{}-{}'.format(


Loading…
Cancel
Save