Yonuts commited on
Commit
66c926c
·
1 Parent(s): 33dc149
Files changed (1) hide show
  1. models/PDNet.py +1 -1
models/PDNet.py CHANGED
@@ -8,7 +8,7 @@ from deepinv.unfolded import unfolded_builder
8
  from deepinv.utils.phantoms import RandomPhantomDataset, SheppLoganDataset
9
  from deepinv.optim.optim_iterators import CPIteration, fStep, gStep
10
  from deepinv.optim import Prior, DataFidelity
11
- from deepinv.utils.tensorlist import TensorList
12
 
13
  from physics.multiscale import MultiScaleLinearPhysics
14
  from models.heads import Heads, Tails, InHead, OutTail, ConvChannels, SNRModule, EquivConvModule, EquivHeads
 
8
  from deepinv.utils.phantoms import RandomPhantomDataset, SheppLoganDataset
9
  from deepinv.optim.optim_iterators import CPIteration, fStep, gStep
10
  from deepinv.optim import Prior, DataFidelity
11
+ from deepinv.utils import TensorList
12
 
13
  from physics.multiscale import MultiScaleLinearPhysics
14
  from models.heads import Heads, Tails, InHead, OutTail, ConvChannels, SNRModule, EquivConvModule, EquivHeads