Add doc strings.
change ResNet to blocks accordingly.
# limitations under the License.
#
-from .ResNet import *
+'''
+This module includes all the prebuild models.
+'''
from .Sequential import Sequential
+from .blocks import ResBlockBN, ResBlock_Caffe, DenseNetBlock
from .caffe_models import model_lenet, model_vgg19, model_resnet50, model_resnet101, model_resnet152
from .layers import *
from .model import Model
from .utils import random_name
-# TODO: Needs docstring
-
def LeNet5(conn, model_name='LENET5',
n_classes=10, n_channels=1, width=28, height=28, scale=1.0 / 255,
random_flip='NONE', random_crop='NONE', offsets=0):
@@ -2247,6 +2248,7 @@ def wide_resnet(conn, model_name='WIDE_RESNET', batch_norm_first=True, depth=2,
return model
+
def DenseNet_Cifar(conn, model_name=None, n_classes=None, conv_channel=16, growth_rate=12,
n_blocks=4, n_cells=4, n_channels=3, width=32, height=32, scale=1,
random_flip='H', random_crop='UNIQUE', offsets=(85, 111, 139)):
