| 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) | |