Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
zizhaozhang authored Dec 16, 2021
1 parent 1b1d7bd commit df79edd
Showing 1 changed file with 22 additions and 3 deletions.
25 changes: 22 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@ L2P is a novel continual learning technique which learns to dynamically prompt a

Code is written by Zifeng Wang. Acknowledgement to https://github.com/google-research/nested-transformer.

This is not an officially supported Google product.

## Enviroment setup
```
pip install -r requirements.txt
Expand Down Expand Up @@ -37,10 +39,27 @@ Note: we run our experiments using 8 V100 GPUs or 4 TPUs, and we specify a per d


## Visualize results
TODO(zifengw): Add instructions and metrics explanation.
We use tensorboard to visualize the result. For example, if the working directory specified to run L2P is `workdir=./cifar100_l2p`, the command to check result is as follows:

```
tensorboard --logdir ./cifar100_l2p
```
Here are the important metrics to keep track of, and their corresponding meanings:

| Metric | Description |
| ----------- | ----------- |
| accuracy_n | Accuracy of the n-th task |
| forgetting | Average forgetting up until the current task |
| avg_acc | Average evaluation accuracy up until the current task |

## Cite
TODO(zifengw): Add bib.


## Cite
```
@inproceedings{wang2021learning,
title={Learning to Prompt for Continual Learning},
author={Zifeng Wang and Zizhao Zhang and Chen-Yu Lee and Han Zhang and Ruoxi Sun and Xiaoqi Ren and Guolong Su and Vincent Perot and Jennifer Dy and Tomas Pfister},
booktitle={arXiv preprint}, # Update this after arxived.
year={2021}
}
```

0 comments on commit df79edd

Please sign in to comment.