o
    }oi                     @   s.   d dl mZ d dlmZ G dd deeZdS )    )Serialization)ModelPTc                       s@   e Zd Z fddZedd Zdd Zdd Zd	d
 Z  Z	S )NerfModelBasec                    s    t  j|d |   || _d S )N)cfg)super__init__save_hyperparameters_cfg)selfr   	__class__ `/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/multimodal/models/nerf/base.pyr      s   
zNerfModelBase.__init__c                 C   s   t | do	t| jS )Nupdate_step)hasattrcallabler   )moduler   r   r   is_module_updatable   s   z!NerfModelBase.is_module_updatablec                 C      d S Nr   )r
   r   r   r   list_available_models      z#NerfModelBase.list_available_modelsc                 C   r   r   r   r
   configr   r   r   setup_training_data    r   z!NerfModelBase.setup_training_datac                 C   r   r   r   r   r   r   r   setup_validation_data#   r   z#NerfModelBase.setup_validation_data)
__name__
__module____qualname__r   staticmethodr   r   r   r   __classcell__r   r   r   r   r      s    
r   N)nemo.core.classes.commonr   nemo.core.classes.modelPTr   r   r   r   r   r   <module>   s   