    def init_modules(self, detection_model_name, workdir):
        """
        Load the AIT module of the detection model, and set the weights.
        """
        mod = Model(os.path.join(workdir, detection_model_name, "test.so"))
        mod.set_many_constants_with_tensors(self.weights)
        mod.fold_constants(sync=True)
        return mod
Load the AIT module of the detection model, and set the weights.
examples/02_detectron2/predictor/predictor.py