mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Remove eval to avoid safety concerns

Browse Source
This commit is contained in:
Ross Wightman 2024-08-07 14:54:13 -07:00
parent ea33d6a74d
commit 49f6e1da96
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
timm/models/rdnet.py
View File

@ -52,6 +52,16 @@ class BlockESE(nn.Module):
return self.layers(x) return self.layers(x)
def _get_block_type(block: str):
block = block.lower().strip()
if block == "block":
return Block
elif block == "blockese":
return BlockESE
else:
assert False, f"Unknown block type ({block})."
class DenseBlock(nn.Module): class DenseBlock(nn.Module):
def __init__( def __init__(
self, self,
@ -80,7 +90,7 @@ class DenseBlock(nn.Module):
self.drop_path = DropPath(drop_path_rate) self.drop_path = DropPath(drop_path_rate)
self.layers = eval(block_type)( self.layers = _get_block_type(block_type)(
in_chs=num_input_features, in_chs=num_input_features,
inter_chs=inter_chs, inter_chs=inter_chs,
out_chs=growth_rate, out_chs=growth_rate,
@ -130,7 +140,7 @@ class RDNet(nn.Module):
global_pool: str = 'avg', # timm option [--gp] global_pool: str = 'avg', # timm option [--gp]
growth_rates: Union[List[int], Tuple[int]] = (64, 104, 128, 128, 128, 128, 224), growth_rates: Union[List[int], Tuple[int]] = (64, 104, 128, 128, 128, 128, 224),
num_blocks_list: Union[List[int], Tuple[int]] = (3, 3, 3, 3, 3, 3, 3), num_blocks_list: Union[List[int], Tuple[int]] = (3, 3, 3, 3, 3, 3, 3),
block_type: Union[List[int], Tuple[int]] = ("Block", "Block", "BlockESE", "BlockESE", "BlockESE", "BlockESE", "BlockESE"), block_type: Union[List[int], Tuple[int]] = ("Block",) * 2 + ("BlockESE",) * 5,
is_downsample_block: Union[List[bool], Tuple[bool]] = (None, True, True, False, False, False, True), is_downsample_block: Union[List[bool], Tuple[bool]] = (None, True, True, False, False, False, True),
bottleneck_width_ratio: float = 4.0, bottleneck_width_ratio: float = 4.0,
transition_compression_ratio: float = 0.5, transition_compression_ratio: float = 0.5,