[Bugfix] Update deprecated method name in load_graph.py (#2769)
Method `dataset.num_labels` has been deprecated and replaced by `dataset.num_classes`.
Updating the method name to avoid runtime warning.
Co-authored-by:
Minjie Wang <wmjlyjemaine@gmail.com>
Showing
Please register or sign in to comment