Skip to content

Commit

Permalink
use 100% width
Browse files Browse the repository at this point in the history
  • Loading branch information
sumn2u committed Jan 8, 2024
1 parent 36a5124 commit 4b6a58f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions paper/paper.md
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,9 @@ Integration of machine learning models with mobile devices presents a promising

The model was trained with Tesla T4 GPU and uses EfficientNetV2 [@tan2021efficientnetv2] model as a base model with addition of agumentation layer. Adam was used as an optmizer with intital learning rate of 0.01. Which was later optmised using [optuna](https://optuna.org/) to create more accurate optimization parameters. The training and validation loss is shown in \autoref{fig:training_vs_val_loss} whereas \autoref{fig:training_vs_val_accuracy} shows training and validation accuracy on the performed experiment[^2].

![Training and Validation loss at different epochs\label{fig:training_vs_val_loss}](training_vs_val_loss.png){width="48%"}
![Training and Validation loss at different epochs\label{fig:training_vs_val_loss}](training_vs_val_loss.png){width="100%"}

![Training and Validation accuracy at different epochs\label{fig:training_vs_val_accuracy}](training_vs_val_accuracy.png){width="48%"}
![Training and Validation accuracy at different epochs\label{fig:training_vs_val_accuracy}](training_vs_val_accuracy.png){width="100%"}


The confusion matix of the modle is shown in \autoref{fig:confusion_matrix}.
Expand Down

0 comments on commit 4b6a58f

Please sign in to comment.