search

microsoft / Graphormer

Graphormer is a general-purpose deep learning backbone for molecular modeling.
MIT License
2k stars 324 forks source link

graph_prediction.py calls the EpochShuffleDataset method with wrong parameters name #131

Closed chrisxu2016 closed 1 year ago

chrisxu2016 commented 1 year ago

https://github.com/microsoft/Graphormer/blob/bdbbe6613a8392688babc825e20b1173ae82d6f7/graphormer/tasks/graph_prediction.py#L215

As the title says, here should be num_samples instead of size

mavisguan commented 1 year ago

Thanks for helping us finding out the bug, we'll fix this soon.