Python KPN_1f_fs64 Examples

Example #1

File: model_io.py Project: gauenk/cl_gen

def load_model_kpn_1f(cfg):
    if cfg.dynamic.frame_size == 128:
        return KPN_1f(color=True,kernel_size=[cfg.kpn_1f_frame_size],burst_length=cfg.input_N,blind_est=True),LossFunc(tensor_grad=~cfg.blind,alpha=1.0)
    elif cfg.dynamic.frame_size == 64:
        return KPN_1f_fs64(color=True,kernel_size=[cfg.kpn_1f_frame_size],burst_length=cfg.input_N,blind_est=True),LossFunc(tensor_grad=~cfg.blind,alpha=1.0)
    elif cfg.dynamic.frame_size == 32:
        return KPN_1f_fs32(color=True,kernel_size=[cfg.kpn_1f_frame_size],burst_length=cfg.input_N,blind_est=True),LossFunc(tensor_grad=~cfg.blind,alpha=1.0)
    else:
        raise KeyError("Uknown frame size [{cfg.dynamic.frame_size}]")

Example #2

def load_model_kpn_1f(cfg):
    if cfg.dynamic.frame_size == 128:
        return KPN_1f(color=True,
                      kernel_size=[9],
                      burst_length=cfg.input_N,
                      blind_est=True), LossFunc(tensor_grad=~cfg.blind,
                                                alpha=1.0)
    else:
        return KPN_1f_fs64(color=True,
                           kernel_size=[9],
                           burst_length=cfg.input_N,
                           blind_est=True), LossFunc(tensor_grad=~cfg.blind,
                                                     alpha=1.0)