pyiqa.models.general_iqa_model ============================== .. py:module:: pyiqa.models.general_iqa_model Module Contents --------------- .. py:class:: GeneralIQAModel(opt) Bases: :py:obj:`pyiqa.models.base_model.BaseModel` General module to train an IQA network. .. py:method:: init_training_settings() .. py:method:: setup_optimizers() .. py:method:: feed_data(data) .. py:method:: net_forward(net) .. py:method:: optimize_parameters(current_iter) .. py:method:: test() .. py:method:: dist_validation(dataloader, current_iter, tb_logger, save_img) .. py:method:: nondist_validation(dataloader, current_iter, tb_logger, save_img) .. py:method:: save(epoch, current_iter, save_net_label='net')