Skip to content

Commit

Permalink
remove embed
Browse files Browse the repository at this point in the history
  • Loading branch information
Richard Zhang authored and Richard Zhang committed Oct 6, 2020
1 parent bfc22d3 commit ed61e9f
Show file tree
Hide file tree
Showing 4 changed files with 3 additions and 5 deletions.
2 changes: 0 additions & 2 deletions antialiased_cnns/alexnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@
import torch.utils.model_zoo as model_zoo
import numpy as np
from antialiased_cnns import *
from IPython import embed

__all__ = ['AlexNet', 'alexnet']

Expand Down Expand Up @@ -153,7 +152,6 @@ def __init__(self, num_classes=1000, filter_size=1):
)

def forward(self, x):
# embed()
x = self.features(x)
x = self.avgpool(x)
x = x.view(x.size(0), 256 * 6 * 6)
Expand Down
1 change: 0 additions & 1 deletion antialiased_cnns/blurpool.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
import numpy as np
import torch.nn as nn
import torch.nn.functional as F
from IPython import embed

class BlurPool(nn.Module):
def __init__(self, channels, pad_type='reflect', filt_size=4, stride=2, pad_off=0):
Expand Down
1 change: 0 additions & 1 deletion antialiased_cnns/resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,6 @@
import torch.nn as nn
import torch.utils.model_zoo as model_zoo
from antialiased_cnns import *
from IPython import embed

__all__ = ['ResNet', 'resnet18', 'resnet34', 'resnet50', 'resnet101',
'resnet152', 'resnext50_32x4d', 'resnext101_32x8d']
Expand Down
4 changes: 3 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
author_email="rizhang@adobe.com",
description="Antialiased models and pooling layer from Zhang. Making Convnets Shift-Invariant Again. ICML 2019.",
long_description=long_description,
long_description_content_type="text/markdown",
long_description_content_type="text/markdown",
url="https://github.com/adobe/antialiased-cnns",
packages=setuptools.find_packages(),
classifiers=[
Expand All @@ -17,3 +17,5 @@
"Operating System :: OS Independent",
],
)

# install_requires=[],

0 comments on commit ed61e9f

Please sign in to comment.