Skip to content

Commit 4f51cf9

Browse files
authored
Merge pull request aws#93 from awslabs/sentiment-gluon
Fix comments
2 parents 7c2974e + 8c9fc7a commit 4f51cf9

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

sagemaker-python-sdk/mxnet_gluon_sentiment/sentiment.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@ def train(current_host, hosts, num_cpus, num_gpus, channel_input_dirs, model_dir
2727
learning_rate = hyperparameters.get('learning_rate', 0.01)
2828
log_interval = hyperparameters.get('log_interval', 1000)
2929
embedding_size = hyperparameters.get('embedding_size', 50)
30-
wd = hyperparameters.get('wd', 0.0001)
3130

3231
if len(hosts) == 1:
3332
kvstore = 'device' if num_gpus > 0 else 'local'
@@ -100,9 +99,7 @@ def train(current_host, hosts, num_cpus, num_gpus, channel_input_dirs, model_dir
10099

101100

102101
class BucketSentenceIter(DataIter):
103-
"""Simple bucketing iterator for language model.
104-
The label at each sequence step is the following token
105-
in the sequence.
102+
"""Simple bucketing iterator for text classification model.
106103
Parameters
107104
----------
108105
sentences : list of list of int

0 commit comments

Comments
 (0)