Train_Mask use - Githubissues

benedekrozemberczki / pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

MIT License

2.61k stars 367 forks source link

Closed he-h closed 1 year ago

he-h commented 1 year ago

Is there anyway I can add train_mask function during the training or can you show an example on how to use these?