From 4d8f856130a4d9e21841ca1f612233245de9e1e7 Mon Sep 17 00:00:00 2001 From: consilium538 Date: Thu, 29 Aug 2019 17:05:16 +0900 Subject: [PATCH] Repeated block block_args.stride int to [int] --- efficientnet_pytorch/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/efficientnet_pytorch/model.py b/efficientnet_pytorch/model.py index 3a42217..ed14b19 100644 --- a/efficientnet_pytorch/model.py +++ b/efficientnet_pytorch/model.py @@ -139,7 +139,7 @@ def __init__(self, blocks_args=None, global_params=None): # The first block needs to take care of stride and filter size increase. self._blocks.append(MBConvBlock(block_args, self._global_params)) if block_args.num_repeat > 1: - block_args = block_args._replace(input_filters=block_args.output_filters, stride=1) + block_args = block_args._replace(input_filters=block_args.output_filters, stride=[1]) for _ in range(block_args.num_repeat - 1): self._blocks.append(MBConvBlock(block_args, self._global_params))