From 2b9fb6396316e11971f2136598eea99a4074b6f0 Mon Sep 17 00:00:00 2001 From: Yuwen Heng <15895948+123mutourener@users.noreply.github.com> Date: Mon, 6 Jun 2022 11:53:22 +0800 Subject: [PATCH] Update README.md --- README.md | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 6e7fe75..d28cdc3 100644 --- a/README.md +++ b/README.md @@ -15,4 +15,12 @@ docker pull 123mutouren/cv:1.0.0 ## Local Material Dataset Please download the original dataset from https://vision.ist.i.kyoto-u.ac.jp/codeanddata/localmatdb/, into the folder dataset/localmatdb. Then you can zip the folder localmatdb since our dataloader assumes the images are zipped. - +## Train DBAT +To train our DBAT, you can use the code below: +``` +python train_sota.py --data-root "./dataset" --batch-size 4 --tag dpglt --gpus 1 --num-nodes 1 --epochs 200 --mode 95 --seed 42 +``` +To test the trained model, you can specify the checkpoint path with the --test option +``` +python train_sota.py --data-root "./dataset" --batch-size 4 --tag dpglt --gpus 1 --num-nodes 1 --epochs 200 --mode 95 --seed 42 --test accuracy/epoch\=126-valid_acc_epoch\=0.87.ckpt +``` -- GitLab