JiaxuanYou / graph-generation

GraphRNN: Generating Realistic Graphs with Deep Auto-regressive Models
MIT License
693 stars 141 forks source link

Covariate Adjustment + Single Graph + CPU Usage #4

Closed jlevy44 closed 5 years ago

jlevy44 commented 5 years ago

I know how to convert these torch tensors to CPU, I am curious though why the user does not have the choice to run CUDA.

Also, I have a custom graph with some covariates, how can I train on these to compare to ERGM? Particularly, what would you recommend for using covariates for graphRNN generation?

I'm just having trouble locating where to input a single graph for this pipeline, as the paper said that single or multiple graphs can be used.