a
    ego                     @   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:\Games\Steam\steamapps\common\wallpaper_engine\dlc\pymidas\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   