Skip to content

Commit f6744f1

Browse files
committed
feat(aug): merge data aug strategy from DeltaFlow project.
1 parent 876cfd4 commit f6744f1

3 files changed

Lines changed: 290 additions & 135 deletions

File tree

conf/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ gradient_clip_val: 5.0
2929
optimizer:
3030
name: Adam # [Adam, AdamW]
3131
lr: 1e-3
32-
loss_fn: seflowLoss # choices: [ff3dLoss, zeroflowLoss, deflowLoss, seflowLoss]
32+
loss_fn: deflowLoss # choices: [ff3dLoss, zeroflowLoss, deflowLoss, seflowLoss]
3333
# add_seloss: {chamfer_dis: 1.0, static_flow_loss: 1.0, dynamic_chamfer_dis: 1.0, cluster_based_pc0pc1: 1.0}
3434
# ssl_label:
3535

0 commit comments

Comments
 (0)