create_jit_net_from_config_and_weight

create_jit_net_from_config_and_weight(config_dict, weight_file)[source]

Creates a JIT Network from config dict and weight file

Parameters:
  • config_dict (dict) – dict containing network configuration
  • weight_file (str) – path to file containing weights
Returns:

jitted network

Return type:

torch.jit.ScriptModule

main

main()[source]