pyiqa.models.general_iqa_model¶

Module Contents¶

class pyiqa.models.general_iqa_model.GeneralIQAModel(opt)[source]¶

Bases: pyiqa.models.base_model.BaseModel

General module to train an IQA network.

init_training_settings()[source]¶

setup_optimizers()[source]¶

feed_data(data)[source]¶

net_forward(net)[source]¶

optimize_parameters(current_iter)[source]¶

test()[source]¶

dist_validation(dataloader, current_iter, tb_logger, save_img)[source]¶

nondist_validation(dataloader, current_iter, tb_logger, save_img)[source]¶

save(epoch, current_iter, save_net_label='net')[source]¶