o
    Wbfo                     @   s    d dl Z G dd de jjZdS )    Nc                   @   s   e Zd Zdd ZdS )	BaseModelc                 C   s2   t j|t dd}d|v r|d }| | dS )zOLoad model from file.

        Args:
            path (str): file path
        cpu)map_location	optimizermodelN)torchloaddeviceload_state_dict)selfpath
parameters r   PC:\Users\Kristjan\AppData\Local\Programs\Python\MiDaS-master\midas\base_model.pyr      s   zBaseModel.loadN)__name__
__module____qualname__r   r   r   r   r   r      s    r   )r   nnModuler   r   r   r   r   <module>   s    