reset default l2 reg value and provide pretrained model

Author: pmixer

README.md

@@ -1,3 +1,12 @@
+
+update: a pretrained model added, pls run the command as below to test its performance:
+
+```
+python main.py --dataset=ml-1m --train_dir=default --dropout_rate=0.2 --device=cuda --state_dict_path='ml-1m_default/TiSASRec.epoch=601.lr=0.001.layer=2.head=1.hidden=50.maxlen=200.pth' --inference_only=true --maxlen=200
+```
+
+---
+
 modified based on [paper author's tensorflow implementation](https://github.com/JiachengLi1995/TiSASRec), switching to PyTorch(v1.6) for simplicity, executable by:
 
 ```python main.py --dataset=ml-1m --train_dir=default --device=cuda```

main.py

@@ -24,7 +24,7 @@ def str2bool(s):
 parser.add_argument('--num_epochs', default=201, type=int)
 parser.add_argument('--num_heads', default=1, type=int)
 parser.add_argument('--dropout_rate', default=0.2, type=float)
-parser.add_argument('--l2_emb', default=0.0, type=float)
+parser.add_argument('--l2_emb', default=0.00005, type=float)
 parser.add_argument('--device', default='cpu', type=str)
 parser.add_argument('--inference_only', default=False, type=str2bool)
 parser.add_argument('--state_dict_path', default=None, type=str)