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

profiling in training script

fixing the dataset checking script

added check data script

fixed imports

updated training and inference script

added setup scripts

added root_gnn_dgl directory