This commit is contained in:
Kohaku-Blueleaf 2024-01-05 16:38:05 +08:00
parent 18ca987c92
commit 44744d6005
1 changed files with 1 additions and 2 deletions

View File

@ -3,7 +3,6 @@ import os
from collections import namedtuple
import enum
import torch
import torch.nn as nn
import torch.nn.functional as F
@ -124,7 +123,7 @@ class NetworkModule:
if isinstance(self.sd_module, nn.Conv2d):
self.ops = F.conv2d
self.extra_kwargs = {
'stride': self.sd_module.stride,
'stride': self.sd_module.stride,
'padding': self.sd_module.padding
}
elif isinstance(self.sd_module, nn.Linear):