thainamhoang commited on
Commit
30aaf9c
1 Parent(s): 23247f8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -8,4 +8,4 @@ license: mit
8
  - Model: I'm using `ResNet50` with `pretrained=True` from `pytorch`. The choice of optimizer is `Adam` with `lr=1e-4`. I had tried `AdamW` before coming back to `Adam`. I also added the `lr_scheduler` to reconfigure the learning rate.
9
  - Number of epochs: I'm using 10 epochs in total. I had tried with 20, however, only the `train_acc` increases by ~1%, so it does not worth trying for an increasing amount of training time.
10
 
11
- [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1FgVFvmjFn-tGk1pzVUTfra6UyafcuPLV?usp=sharing)
 
8
  - Model: I'm using `ResNet50` with `pretrained=True` from `pytorch`. The choice of optimizer is `Adam` with `lr=1e-4`. I had tried `AdamW` before coming back to `Adam`. I also added the `lr_scheduler` to reconfigure the learning rate.
9
  - Number of epochs: I'm using 10 epochs in total. I had tried with 20, however, only the `train_acc` increases by ~1%, so it does not worth trying for an increasing amount of training time.
10
 
11
+ Model: [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1FgVFvmjFn-tGk1pzVUTfra6UyafcuPLV?usp=sharing)