Merge branch 'main' into dev

Author: meta-tabchen

README.md

@@ -65,10 +65,10 @@ pip install -U pykt-toolkit -i  https://pypi.python.org/simple
 We now have a [paper](https://arxiv.org/abs/2206.11460?context=cs.CY) you can cite for the our pyKT library:
 
 ```bibtex
-@article{liu2022pykt,
+@inproceedings{liupykt2022,
   title={pyKT: A Python Library to Benchmark Deep Learning based Knowledge Tracing Models},
   author={Liu, Zitao and Liu, Qiongqiong and Chen, Jiahao and Huang, Shuyan and Tang, Jiliang and Luo, Weiqi},
-  journal={arXiv preprint arXiv:2206.11460},
+  booktitle={Thirty-sixth Conference on Neural Information Processing Systems Datasets and Benchmarks Track},
   year={2022}
 }
-```
+```

examples/wandb_train.py

@@ -67,7 +67,7 @@ def main(params):
     print(dataset_name, model_name, data_config, fold, batch_size)
 
     debug_print(text="init_dataset",fuc_name="main")
-    train_loader, valid_loader = init_dataset4train(dataset_name, model_name, data_config, fold, batch_size)
+    train_loader, valid_loader, *_ = init_dataset4train(dataset_name, model_name, data_config, fold, batch_size)
 
     params_str = "_".join([str(v) for k,v in params.items() if not k in ['other_config']])
 

pykt/datasets/data_loader.py

@@ -11,6 +11,11 @@
     from torch import FloatTensor, LongTensor
 import numpy as np
 
+if torch.cuda.is_available():
+    from torch.cuda import FloatTensor, LongTensor
+else:
+    from torch import FloatTensor, LongTensor
+
 class KTDataset(Dataset):
     """Dataset for KT
         can use to init dataset for: (for models except dkt_forget)