dc9aafd83851f7c55c6fe61702e281856ec023ca,models/modules/shift_unet.py,InceptionUnetSkipConnectionBlock,__init__,#InceptionUnetSkipConnectionBlock#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,387

Before Change

        // As the downconv layer is outer_nc in and inner_nc out.
        // So the shift define like this:
        shift = InnerSoftShiftTriple(opt.threshold, opt.fixed_mask, opt.shift_sz, opt.stride, opt.mask_thred, opt.triple_weight)

        shift.set_mask(mask_global, 3, opt.threshold)
        shift_list.append(shift)

        // Add latent constraint
        // Then add the constraint to the constrain layer list!
        innerCosBefore = InnerCos(strength=opt.strength, skip=opt.skip)

After Change

        if shift_layer:
            // As the downconv layer is outer_nc in and inner_nc out.
            // So the shift define like this:
            shift = InnerSoftShiftTriple(opt.threshold, opt.fixed_mask, opt.shift_sz, opt.stride, opt.mask_thred, opt.triple_weight)

            shift.set_mask(mask_global, 3, opt.threshold)
            shift_list.append(shift)

            // Add latent constraint
            // Then add the constraint to the constrain layer list!
            innerCosBefore = InnerCos(strength=opt.strength, skip=opt.skip)

Instances