test round dim questions - Githubissues

datawhalechina / torch-rechub

A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.

MIT License

374 stars 69 forks source link

test round dim questions #49

Open caofeng0528 opened 1 year ago

caofeng0528 commented 1 year ago

https://github.com/datawhalechina/torch-rechub/blob/552bf235baf206a2101dc95be9f199e8f2d11c84/torch_rechub/trainers/mtl_trainer.py#L177 When the model is PLE, why is the input time 100x16 and only the first two columns used for test