Update train.py

main
PENG Bo 4 years ago committed by GitHub
parent 34fa2ec81b
commit 689a6a924d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -125,7 +125,7 @@ NUM_OF_RUNS = 5
LENGTH_OF_EACH = 300
for run in range(NUM_OF_RUNS):
context = "it was"
context = "\n"
if model_level == 'word':
x = np.array([train_dataset.stoi[s] for s in context.strip().lower().split(' ')], dtype=np.int64)

Loading…
Cancel
Save