message passing and rename this metadata
Line: 58
|
} |
|
std::unordered_map<std::string, std::string> metadata = {{"type", ""}}; |
|
module = torch::jit::load(model, device, metadata); |
|
// TODO: This should be fixed after implement api to decide whether need to |
|
// message passing and rename this metadata |
|
if (metadata["type"] == "dpa2") { |
|
do_message_passing = 1; |
|
} else { |
|
do_message_passing = 0; |
|
} |
|
torch::jit::FusionStrategy strategy; |