Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

LORE 训练模型推理 #182

Open
cqray1990 opened this issue Oct 14, 2024 · 0 comments
Open

LORE 训练模型推理 #182

cqray1990 opened this issue Oct 14, 2024 · 0 comments

Comments

@cqray1990
Copy link

训练命令:
python main.py --task ctdet
--dataset_name table
--exp_id training_wtw
--dataset_name WTW
--image_dir /train/images
--wiz_4ps
--wiz_stacking
--wiz_pairloss
--tsfm_layers 3
--stacking_layers 3
--batch_size 32
--master_batch 6
--arch dla_34
--lr 1e-4
--K 500
--MK 1000
--num_epochs 100
--lr_step '70, 90'
--gpus 0,1,2,3
--num_workers 16
--val_intervals 10

得到训练模型推理命令:
python demo.py --task ctdet
--dataset table
--demo /home/user/Docllm/docprj/AdvancedLiterateMachinery/DocumentUnderstanding/LORE-TSR/input_images/wired
--demo_name demo_wired
--debug 1
--arch dla_34
--K 3000
--MK 5000
--tsfm_layers 4
--stacking_layers 4
--gpus 0
--wiz_4ps
--wiz_detect
--wiz_rev
--wiz_stacking
--convert_onnx 1
--vis_thresh_corner 0.3
--vis_thresh 0.20
--scores_thresh 0.2
--nms
--demo_dir ../visualization_wired/
--load_model /exp/ctdet/training_wtw/model_last.pth
--load_processor /exp/ctdet/training_wtw/processor_last.pth
为什么会提示以错误呢:用仓库的给的模型没有提示这种参数找不到的

No param stacker.tsfm.encoder.layers.3.norm_1.alpha.
No param stacker.tsfm.encoder.layers.3.norm_1.bias.
No param stacker.tsfm.encoder.layers.3.norm_2.alpha.
No param stacker.tsfm.encoder.layers.3.norm_2.bias.
No param stacker.tsfm.encoder.layers.3.attn.q_linear.weight.
No param stacker.tsfm.encoder.layers.3.attn.q_linear.bias.
No param stacker.tsfm.encoder.layers.3.attn.v_linear.weight.
No param stacker.tsfm.encoder.layers.3.attn.v_linear.bias.
No param stacker.tsfm.encoder.layers.3.attn.k_linear.weight.
No param stacker.tsfm.encoder.layers.3.attn.k_linear.bias.
No param stacker.tsfm.encoder.layers.3.attn.out.weight.
No param stacker.tsfm.encoder.layers.3.attn.out.bias.
No param stacker.tsfm.encoder.layers.3.ff.linear_1.weight.
No param stacker.tsfm.encoder.layers.3.ff.linear_1.bias.
No param stacker.tsfm.encoder.layers.3.ff.linear_2.weight.
No param stacker.tsfm.encoder.layers.3.ff.linear_2.bias.
No param tsfm_axis.encoder.layers.3.norm_1.alpha.
No param tsfm_axis.encoder.layers.3.norm_1.bias.
No param tsfm_axis.encoder.layers.3.norm_2.alpha.
No param tsfm_axis.encoder.layers.3.norm_2.bias.
No param tsfm_axis.encoder.layers.3.attn.q_linear.weight.
No param tsfm_axis.encoder.layers.3.attn.q_linear.bias.
No param tsfm_axis.encoder.layers.3.attn.v_linear.weight.
No param tsfm_axis.encoder.layers.3.attn.v_linear.bias.
No param tsfm_axis.encoder.layers.3.attn.k_linear.weight.
No param tsfm_axis.encoder.layers.3.attn.k_linear.bias.
No param tsfm_axis.encoder.layers.3.attn.out.weight.
No param tsfm_axis.encoder.layers.3.attn.out.bias.
No param tsfm_axis.encoder.layers.3.ff.linear_1.weight.
No param tsfm_axis.encoder.layers.3.ff.linear_1.bias.
No param tsfm_axis.encoder.layers.3.ff.linear_2.weight.
No param tsfm_axis.encoder.layers.3.ff.linear_2.bias.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant