hofee/nbv_reconstruction
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
hofee 2024-10-29 16:59:03 +00:00
parent b3a7650d3e
commit 420e9c97bd
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
configs/server/server_train_config.yaml
View File

@ -39,7 +39,7 @@ dataset:
type: train type: train
cache: True cache: True
ratio: 1 ratio: 1
batch_size: 64 batch_size: 32
num_workers: 16 num_workers: 16
pts_num: 8192 pts_num: 8192
load_from_preprocess: True load_from_preprocess: True
@ -55,7 +55,7 @@ dataset:
eval_list: eval_list:
- pose_diff - pose_diff
ratio: 1 ratio: 1
batch_size: 64 batch_size: 32
num_workers: 12 num_workers: 12
pts_num: 8192 pts_num: 8192
load_from_preprocess: True load_from_preprocess: True
@ -71,7 +71,7 @@ dataset:
eval_list: eval_list:
- pose_diff - pose_diff
ratio: 1 ratio: 1
batch_size: 64 batch_size: 32
num_workers: 12 num_workers: 12
pts_num: 8192 pts_num: 8192
load_from_preprocess: True load_from_preprocess: True
@ -92,12 +92,12 @@ module:
pointnet_encoder: pointnet_encoder:
in_dim: 3 in_dim: 3
out_dim: 1024 out_dim: 512
global_feat: True global_feat: True
feature_transform: False feature_transform: False
transformer_seq_encoder: transformer_seq_encoder:
embed_dim: 1280 embed_dim: 768
num_heads: 4 num_heads: 4
ffn_dim: 256 ffn_dim: 256
num_layers: 3 num_layers: 3