INNER CODE UNIT · Python

build_model_func

OpenMeshLab/MeshXL · main.py:135

def build_model_func(args):
    model_module = importlib.import_module(f'models.{args.model}.get_model')
    model = model_module.get_model(args)
    return model


def main(args):
    
    np.random.seed(args.seed)
    torch.cuda.manual_seed_all(args.seed)
    
    if args.checkpoint_dir is not None:
        pass
    elif args.test_ckpt is not None:
        # if not define the checkpoint-dir, set to the test checkpoint folder as default
        args.checkpoint_dir = os.path.dirname(args.test_ckpt)
        print(f'testing directory: {args.checkpoint_dir}')
    else:

View source record →