freexiaochuan/captcha
best.pth model = CaptchaCNNTransformer( img_h=32, img_w=128, dim=256, depth=6, heads=4, num_classes=num_classes, channels=1, dropout=0.2, ).to(device)
0192
- best.pth
model = CaptchaCNNTransformer(
img_h=32,
img_w=128,
dim=256,
depth=6,
heads=4,
num_classes=num_classes,
channels=1,
dropout=0.2,
).to(device)