Downloads · 30 days
0
resnet50_fatima_challenge
thainamhoang/resnet50_fatima_challenge
resnet50_fatima_challenge is a machine learning model from thainamhoang. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
- Dataset: I'm using the catsvsdogs dataset from kaggle instead of huggingface datasets since catsvsdogs in datasets have checksum error. The original dataset is from Microsoft. However, some files are corrupted and i…
Access
Public
Updated Apr 8, 2022
Repo size
—
Likes
0
Public
Repo makeup
Click a slice to open those files.
Other1.2 KB · 57%
At a glance
- License
- mit
- Access
- Public
- Created
- Apr 8, 2022
- Updated
- Apr 8, 2022
- SHA
- 30aaf9c6
Model card
From the Hugging Face model README
Fatima Challenge
- Dataset: I'm using the
cats_vs_dogsdataset from kaggle instead of huggingfacedatasetssincecats_vs_dogsindatasetshave checksum error. The original dataset is from Microsoft. However, some files are corrupted and it hasn't been divided intotrainandtestfolders. - Model: I'm using
ResNet50withpretrained=Truefrompytorch. The choice of optimizer isAdamwithlr=1e-4. I had triedAdamWbefore coming back toAdam. I also added thelr_schedulerto reconfigure the learning rate. - Number of epochs: I'm using 10 epochs in total. I had tried with 20, however, only the
train_accincreases by ~1%, so it does not worth trying for an increasing amount of training time.
Model: 
- License
- mit
- Created
- Apr 8, 2022
- Updated
- Apr 8, 2022