pip install jovian --upgrade -q

import os
import torch
import pandas as pd
import numpy as np
import jovian
from torch.utils.data import Dataset, random_split, DataLoader
from PIL import Image
import torchvision.models as models
import matplotlib.pyplot as plt
import torchvision.transforms as transforms
from sklearn.metrics import f1_score
import torch.nn.functional as F
import torch.nn as nn
from torchvision.utils import make_grid
from tqdm.notebook import tqdm
%matplotlib inline

project_name="HumanProtein-classification"
best_score = 0

print(f"Number of train images: {len(os.listdir('/kaggle/input/jovian-pytorch-z2g/Human protein atlas/train'))}")

print(f"Number of test images: {len(os.listdir('/kaggle/input/jovian-pytorch-z2g/Human protein atlas/test'))}")