@Nahil_Sobh I posted the code on my github account you can see the performance there.
  
      github.com
  
  
    narayana8799/Pneumonia-Detection-using-Pytorch/blob/master/Pneumonia Detection.ipynb 3
{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "764e6867-8a51-4090-b806-07a176c73fe6",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "_CudaDeviceProperties(name='NVIDIA GeForce GTX 1050 Ti', major=6, minor=1, total_memory=4096MB, multi_processor_count=6)"
      ]
     },
     "execution_count": 1,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [


  This file has been truncated. show original