Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
  • Loading branch information
NielsRogge and sgugger committed Jan 21, 2022
1 parent 900ae8d commit e524e0e
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 6 deletions.
4 changes: 2 additions & 2 deletions docs/source/model_doc/convnext.mdx
Expand Up @@ -29,9 +29,9 @@ and outperforming Swin Transformers on COCO detection and ADE20K segmentation, w

Tips:

- One can use the ['AutoFeatureExtractor`] API to prepare images for the model.
- One can use the [`AutoFeatureExtractor`] API to prepare images for the model.

This model was contributed by [nielsr](<https://huggingface.co/nielsr). The original code can be found [here](https://github.com/facebookresearch/ConvNeXt).
This model was contributed by [nielsr](https://huggingface.co/nielsr). The original code can be found [here](https://github.com/facebookresearch/ConvNeXt).

## ConvNextConfig

Expand Down
2 changes: 0 additions & 2 deletions src/transformers/__init__.py
Expand Up @@ -634,7 +634,6 @@
_import_structure["modeling_utils"] = ["Conv1D", "PreTrainedModel", "apply_chunking_to_forward", "prune_layer"]

# PyTorch models structure

_import_structure["models.albert"].extend(
[
"ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
Expand Down Expand Up @@ -2671,7 +2670,6 @@
from .utils.dummy_pytorch_quantization_and_torch_objects import *

if is_torch_available():

# Benchmarks
from .benchmark.benchmark import PyTorchBenchmark
from .benchmark.benchmark_args import PyTorchBenchmarkArguments
Expand Down
4 changes: 2 additions & 2 deletions src/transformers/models/convnext/configuration_convnext.py
Expand Up @@ -28,7 +28,7 @@

class ConvNextConfig(PretrainedConfig):
r"""
This is the configuration class to store the configuration of a [`~ConvNextModel`]. It is used to instantiate an
This is the configuration class to store the configuration of a [`ConvNextModel`]. It is used to instantiate an
ConvNeXT model according to the specified arguments, defining the model architecture. Instantiating a configuration
with the defaults will yield a similar configuration to that of the ConvNeXT
[facebook/convnext-tiny-224](https://huggingface.co/facebook/convnext-tiny-224) architecture.
Expand All @@ -37,7 +37,7 @@ class ConvNextConfig(PretrainedConfig):
documentation from [`PretrainedConfig`] for more information.
Args:
num_channels (`int`, *optional*, defaults to `3`):
num_channels (`int`, *optional*, defaults to 3):
The number of input channels.
num_stages (`int`, optional, defaults to 4):
The number of stages in the model.
Expand Down

0 comments on commit e524e0e

Please sign in to comment.