xinlei55555
/

ViTPain

Model card Files Files and versions

xinlei55555 commited on Feb 16

Commit

32516f2

·

verified ·

1 Parent(s): ea41cae

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -47,7 +47,7 @@ model.eval()
 ## Fine-tuning on UNBC-McMaster
-Use `--au_loss_weight 0.1` when fine-tuning on UNBC (vs 1.0 for synthetic). See the main repo for full training scripts.
 ## Citation

 ## Fine-tuning on UNBC-McMaster
+The ViTPain checkpoints in this repository are the results of pretraining on the 3DPain dataset. A fine-tuned version on the UNBC-McMaster dataset will be released shortly.
 ## Citation