from ml_lib.utils.config import Config
from models import *


class ThisConfig(Config):

    @property
    def _model_map(self):
        return dict(PN2=PointNet2)