dmlc--dgl
Tianyue Cao
05aca98d6a
* upload PSS * upload PSS * upload PSS * pss code reformat * fix bug * update README * update train bash * remove vit * update README * delete InfoPlotter * delete Smooth_AP_loss.py * update README * update README Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com>
9 行
294 B
Bash
9 行
294 B
Bash
python Smooth_AP/src/evaluate_model.py \
|
|
--dataset Inaturalist \
|
|
--bs 384 \
|
|
--source_path ~/code/Smooth_AP/data/ --embed_dim 128 \
|
|
--resume $CHECKPOINT_PATH \
|
|
--class_num 948 --loss smoothap \
|
|
--trainset lin_train_set1.txt \
|
|
--testset Inaturalist_test_set1.txt \
|
|
--linsize 29011 --uinsize 18403 |