Can I ask what are the parameters for running REDDIT-MULTI-12K?
I ran into memory error when trying:
python -m train --bmname=REDDIT-MULTI-12K --assign-ratio=0.1 --hidden-dim=5 --output-dim=11 --cuda=0 --batch-size=10 --num-gc-layers=1 --num-classes=11 --method=soft-assign
thanks!