Align task API and add FunCaptcha support
This commit is contained in:
@@ -22,7 +22,7 @@ class CaptchaClassifier(nn.Module):
|
||||
→ 全局平均池化 → 全连接 → 输出类别数。
|
||||
"""
|
||||
|
||||
def __init__(self, num_types: int = 3):
|
||||
def __init__(self, num_types: int = 5):
|
||||
super().__init__()
|
||||
self.num_types = num_types
|
||||
|
||||
|
||||
Reference in New Issue
Block a user