Skip to content

Commit c558f3f

Browse files
committed
update readme
1 parent e05e258 commit c558f3f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

classification/AntBee/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,4 +46,4 @@ The obtained accuracy by default setting should be around 0.9412, and the AUC wi
4646
![roc](./picture/roc.png)
4747

4848
## Finetuning the last layer of resnet18
49-
Similarly to the above example, we further try to only finetune the last layer of resnet18 for the same classification task. Use a different configure file `config/train_test_ce2.cfg` for training and testing, where `update_layers = -1` in the `network` section means updating the last layer only. Edit `config/evaluation.cfg` accordinly for evaluation. The iteration number obtained the highest accuracy on the validation set was 400, and the accuracy will be around 0.9543. The AUC will be 0.981
49+
Similarly to the above example, we further try to only finetune the last layer of resnet18 for the same classification task. Use a different configure file `config/train_test_ce2.cfg` for training and testing, where `update_layers = -1` in the `network` section means updating the last layer only. Edit `config/evaluation.cfg` accordinly for evaluation. The iteration number obtained the highest accuracy on the validation set was 400 in our testing machine, and the accuracy was around 0.9543. The AUC was 0.981.

0 commit comments

Comments
 (0)