Sean McLeish

Graph_train.py --mem flag

4

Good Afternoon, The [--mem](https://github.com/yilundu/irem_code_release/blob/master/graph_train.py#L140) flag in graph_train.py, does not appear to be used. Only being called: 1) When [declaring iterativeFC](https://github.com/yilundu/irem_code_release/blob/master/graph_train.py#L251) 2) When [declaring GraphEBM](https://github.com/yilundu/irem_code_release/blob/master/graph_train.py#L253) However, it is not used in...

Taking the command line input for lr as it was previously hardcoded so the argument did not change its value. Also removing some unneeded imports.

Adding Huggingface Datasets `.from_generator` support to allow for CLRS-Text datasets to be easily converted to huggingface format.