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