Problem when importing using function "relay.frontend.from_onnx"


#1

Hello, I’m seeking for help of a problem. When using the below code:

“sym, params = relay.frontend.from_onnx(onnx_model, shape_dict)”

“sym”, which represents the NN architecture, is derived. However, when running on different environments, sym with different structure will be generated.

(1) When running on server Linux version 4.4.0-116-generic, dir(sym) = [‘checked_type’, ‘attrs’, ‘body’, ‘params’, ‘ret_type’, ‘span’, ‘type_params’]
(2) When running on mac/virtual machine Ubuntu 16.04.5 LTS, dir(sym) = [‘entry_func’, ‘functions’, ‘global_type_var_map_’, ‘global_var_map_’, ‘type_definitions’]

The versions of tvm are both 0.6.dev.

Why will the outputs be different with the same code? And how to always generate the (1) type output (“sym”)?

This is important to me because I need to extract info from sym


#2

Looks like you have different versions. (2) is a recent change of the frontend API that returns relay.Module


#3

Got it, thank you for answering!