提交 5492c10a 编写于 作者: L LDOUBLEV

fix comments

上级 c7131fec
......@@ -4,12 +4,11 @@ Global:
epoch_num: 500
log_smooth_window: 20
print_batch_step: 10
save_model_dir: ./output/ch_PP-OCR_V3/
save_model_dir: ./output/ch_PP-OCR_v3_det/
save_epoch_step: 100
eval_batch_step:
- 0
- 400
load_static_weights: false
cal_metric_during_train: false
pretrained_model: null
checkpoints: null
......@@ -137,6 +136,7 @@ Train:
img_mode: BGR
channel_first: false
- DetLabelEncode: null
- CopyPaste:
- IaaAugment:
augmenter_args:
- type: Fliplr
......
......@@ -4,12 +4,11 @@ Global:
epoch_num: 500
log_smooth_window: 20
print_batch_step: 10
save_model_dir: ./output/ch_PP-OCR_V3/
save_model_dir: ./output/ch_PP-OCR_V3_det/
save_epoch_step: 100
eval_batch_step:
- 0
- 400
load_static_weights: false
cal_metric_during_train: false
pretrained_model: null
checkpoints: null
......
......@@ -66,7 +66,7 @@ class Head(nn.Layer):
self.conv3 = nn.Conv2DTranspose(
in_channels=in_channels // 4,
out_channels=1,
kernel_size=kernel_size[1],
kernel_size=kernel_size[2],
stride=2,
weight_attr=ParamAttr(
initializer=paddle.nn.initializer.KaimingUniform()),
......
......@@ -20,7 +20,7 @@ import paddle
from paddle import nn
import paddle.nn.functional as F
from paddle import ParamAttr
import os
__dir__ = os.path.dirname(os.path.abspath(__file__))
sys.path.append(__dir__)
sys.path.insert(0, os.path.abspath(os.path.join(__dir__, '../../..')))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册