Skip to content

Commit f625e0e

Browse files
authored
Update recommender.py
1 parent 1cbab4f commit f625e0e

File tree

1 file changed

+1
-1
lines changed
  • introduction_to_applying_machine_learning/gluon_recommender_system

1 file changed

+1
-1
lines changed

introduction_to_applying_machine_learning/gluon_recommender_system/recommender.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def train(channel_input_dirs, hyperparameters, hosts, num_gpus, **kwargs):
4545
num_emb=num_embeddings,
4646
dropout_p=0.5)
4747

48-
net.collect_params().initialize(mx.init.Xavier(magnitude=2.24),
48+
net.collect_params().initialize(mx.init.Xavier(magnitude=60),
4949
ctx=ctx,
5050
force_reinit=True)
5151
net.hybridize()

0 commit comments

Comments
 (0)