diffpool icon indicating copy to clipboard operation
diffpool copied to clipboard

Parameters for REDDIT-MULTI-12K dataset

Open winniehsuanyuan opened this issue 5 years ago • 0 comments

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!

winniehsuanyuan avatar Oct 28 '20 16:10 winniehsuanyuan