steffen/ml_lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Templates finalized

Browse Source
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

2
modules/blocks.py

@ -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')

2
modules/model_parts.py

@ -4,7 +4,7 @@
import torch
from torch import nn
from ml_lib.modules.utils import ShapeMixin
from modules.utils import ShapeMixin
class Generator(nn.Module):

2
modules/utils.py

@ -10,7 +10,7 @@ import pytorch_lightning as pl
# Utility - Modules
###################
from ml_lib.utils.model_io import ModelParameters
from utils.model_io import ModelParameters
class ShapeMixin: