Skip to content

Commit efe9581

Browse files
committed
docs(conf): update conf and env file for local tensorboard.
1 parent 9c465a2 commit efe9581

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

conf/config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ gradient_clip_val: 5.0
2727

2828
# optimizer ==> Adam
2929
lr: 2e-6
30-
loss_fn: seflowLoss # choices: [ff3dLoss, zeroflowLoss, deflowLoss, seflowLoss]
30+
loss_fn: deflowLoss # choices: [ff3dLoss, zeroflowLoss, deflowLoss, seflowLoss]
3131
add_seloss: # {chamfer_dis: 1.0, static_flow_loss: 1.0, dynamic_chamfer_dis: 1.0, cluster_based_pc0pc1: 1.0}
3232

3333
# log settings
@@ -39,4 +39,4 @@ save_top_model: 5 # top_k model will be saved.
3939
# -----> Model and Task Parameters
4040
voxel_size: [0.2, 0.2, 6]
4141
point_cloud_range: [-51.2, -51.2, -3, 51.2, 51.2, 3]
42-
num_frames: 2
42+
num_frames: 2

environment.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ dependencies:
1010
- nvidia/label/cuda-11.7.0::cuda
1111
- lightning==2.0.1
1212
- mkl==2024.0.0
13+
- tensorboard
1314
- numba
1415
- numpy
1516
- pandas

0 commit comments

Comments
 (0)