metadata
license: mit
Fatima Challenge
- Dataset: I'm using the
cats_vs_dogs
dataset from kaggle instead of huggingfacedatasets
sincecats_vs_dogs
indatasets
have checksum error. The original dataset is from Microsoft. However, some files are corrupted and it hasn't been divided intotrain
andtest
folders. - Model: I'm using
ResNet50
withpretrained=True
frompytorch
. The choice of optimizer isAdam
withlr=1e-4
. I had triedAdamW
before coming back toAdam
. I also added thelr_scheduler
to reconfigure the learning rate. - 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.