Templates finalized

This commit is contained in:
Steffen Illium
2020-05-17 22:05:20 +02:00
parent 398504aee6
commit fc93f71608
20 changed files with 459 additions and 166 deletions

@ -5,7 +5,7 @@ import warnings
from torch import nn
from ml_lib.modules.utils import AutoPad, Interpolate, ShapeMixin, F_x, Flatten
from modules.utils import AutoPad, Interpolate, ShapeMixin, F_x, Flatten
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')