Skip to content
This repository has been archived by the owner on Jun 22, 2021. It is now read-only.

CyberZHG/keras-drop-connect

Repository files navigation

Keras Drop-Connect

Travis Coverage

This repo is outdated and will no longer be maintained.

Install

pip install git+https://github.com/cyberzhg/keras-drop-connect.git

Usage

Set drop connect rate for all weights:

import keras
from keras_drop_connect import DropConnect

DropConnect(
    layer=keras.layers.Dense(units=10),
    rate=0.2,
)

Set drop connect rate with a dict:

import keras
from keras_drop_connect import DropConnect

DropConnect(
    layer=keras.layers.Dense(units=10),
    rate={'kernel': 0.2},
)