Gpen-bfr-2048.pth
import torch import torch.nn as nn