Fixing bug where prep_data needs to be called twice. Allowing for different hidden layer sizes

profiling in training script

fixed imports

updated training and inference script

added setup scripts

added root_gnn_dgl directory