Skip to content

Commit

Permalink
Merge pull request #68 from wandering007/patch-1
Browse files Browse the repository at this point in the history 
fix relative import
  • Loading branch information
@fitsumreda
fitsumreda authored Aug 21, 2018
2 parents 4834b4b + 44c8693 commit 3d6db9f
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions models.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,15 +5,15 @@
import math
import numpy as np

from .networks.resample2d_package.modules.resample2d import Resample2d
from .networks.channelnorm_package.modules.channelnorm import ChannelNorm
from networks.resample2d_package.modules.resample2d import Resample2d
from networks.channelnorm_package.modules.channelnorm import ChannelNorm

from .networks import FlowNetC
from .networks import FlowNetS
from .networks import FlowNetSD
from .networks import FlowNetFusion
from networks import FlowNetC
from networks import FlowNetS
from networks import FlowNetSD
from networks import FlowNetFusion

from .networks.submodules import *
from networks.submodules import *
'Parameter count = 162,518,834'

class FlowNet2(nn.Module):
Expand Down

0 comments on commit 3d6db9f

Please sign in to comment.