evpconfig.py

from transformers import PretrainedConfig

class EVPConfig(PretrainedConfig):
    model_type = "EVP"
    def __init__(
            self,
            **kwargs,
                ):
        self.num_deconv = 3
        self.num_filters = [32,32,32]
        self.deconv_kernels = [2,2,2]
        self.dataset = 'nyudepthv2'
        self.max_depth = 10
        self.min_depth_eval = 1e-3
        super().__init__(**kwargs)