Skip to content
This repository was archived by the owner on Jan 1, 2021. It is now read-only.

Commit f797d1a

Browse files
committed
updated config
1 parent 97f9b9e commit f797d1a

2 files changed

Lines changed: 5 additions & 2 deletions

File tree

.gitignore

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,5 @@ examples/checkpoints/*
1010
examples/chatbot/processed/*
1111
examples/chatbot/checkpoints/*
1212
examples/chatbot/data_analysis.py
13+
14+
assignments/chatbot/processed/*

assignments/chatbot/config.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ class CS 20SI: "TensorFlow for Deep Learning Research"
1616
"""
1717

1818
# parameters for processing the dataset
19-
DATA_PATH = 'data/cornell movie-dialogs corpus'
19+
DATA_PATH = '/Users/Chip/data/cornell movie-dialogs corpus'
2020
CONVO_FILE = 'movie_conversations.txt'
2121
LINE_FILE = 'movie_lines.txt'
2222
OUTPUT_FILE = 'output_convo.txt'
@@ -54,7 +54,8 @@ class CS 20SI: "TensorFlow for Deep Learning Research"
5454
BATCH_SIZE = 64
5555

5656
LR = 0.5
57-
LR_DECAY_FACTOR = 0.99
5857
MAX_GRAD_NORM = 5.0
5958

6059
NUM_SAMPLES = 512
60+
ENC_VOCAB = 24414
61+
DEC_VOCAB = 24691

0 commit comments

Comments
 (0)