Skip to content

Commit

Permalink
Add GL-DPN models into list
Browse files Browse the repository at this point in the history
  • Loading branch information
osmr committed Aug 28, 2018
1 parent 752e093 commit 5f6960f
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 2 deletions.
2 changes: 1 addition & 1 deletion gluon/models/dpn.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
import os
from mxnet import cpu
from mxnet.gluon import nn, HybridBlock
from common import conv1x1, DualPathSequential
from .common import conv1x1, DualPathSequential


class GlobalAvgMaxPool2D(HybridBlock):
Expand Down
7 changes: 7 additions & 0 deletions gluon/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
from .models.mobilenet import *
from .models.mobilenetv2 import *
from .models.nasnet import *
from .models.dpn import *

__all__ = ['get_model']

Expand Down Expand Up @@ -156,6 +157,12 @@
'mobilenetv2_wd4': mobilenetv2_wd4,

'nasnet_a_mobile': nasnet_a_mobile,

'dpn68': dpn68,
'dpn68b': dpn68b,
'dpn98': dpn98,
'dpn107': dpn107,
'dpn131': dpn131,
}


Expand Down
2 changes: 1 addition & 1 deletion pytorch/models/dpn.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import torch
import torch.nn as nn
import torch.nn.init as init
from common import conv1x1, DualPathSequential
from .common import conv1x1, DualPathSequential


class GlobalAvgMaxPool2D(nn.Module):
Expand Down

0 comments on commit 5f6960f

Please sign in to comment.