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

fingerprinted now should work correctly

Browse Source
This commit is contained in:
Si11ium
2020-05-19 08:33:04 +02:00
parent e423d6fe31
commit 206aca10b3
7 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules/blocks.py
View File

@ -1,11 +1,13 @@
from pathlib import Path
from typing import Union
import torch
import warnings
from torch import nn
from modules.utils import AutoPad, Interpolate, ShapeMixin, F_x, Flatten
import sys
sys.path.append(str(Path(__file__).parent))
from .util import AutoPad, Interpolate, ShapeMixin, F_x, Flatten
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')