mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

remote unused matplotlib import

pull/653/head
Aman Arora 2021-05-20 23:42:42 +00:00
parent 40c506ba1e
commit 8b1f2e8e1f
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/convit.py
View File

@ -9,13 +9,11 @@ import torch.nn.functional as F
from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
from .helpers import build_model_with_cfg from .helpers import build_model_with_cfg
from timm.models.helpers import load_pretrained
from timm.models.layers import DropPath, to_2tuple, trunc_normal_ from timm.models.layers import DropPath, to_2tuple, trunc_normal_
from timm.models.registry import register_model from timm.models.registry import register_model
import torch import torch
import torch.nn as nn import torch.nn as nn
import matplotlib.pyplot as plt
def _cfg(url='', **kwargs): def _cfg(url='', **kwargs):