diff --git a/network/glue.py b/network/glue.py index 47fd915..855be48 100644 --- a/network/glue.py +++ b/network/glue.py @@ -1,6 +1,6 @@ # Glue layer for transforming whole pictures into 384x384 sequence for encoder input from dataclasses import dataclass -from itertools import product +from itertools import product # product library import torch import torch.nn as nn