Commit 44744d60 authored by Kohaku-Blueleaf's avatar Kohaku-Blueleaf

linting

parent 18ca987c
......@@ -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):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment