Example:
I have a data set size of 801
60% of my data set is used for training so approx 481 size of my training dataset.
dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=15,shuffle=True, num_workers=15)for x in ['train', 'val']}
Above statement will fetch 15 set(mini batch) of data from the training dataset.
so in order to complete training on the entire data set 481/15 approx= 32
so number of epochs required to complete training over the entire dataset = 32 given be the statement below.
model_ft = train_model(net,criterion,optimizer_ft,exp_lr_scheduler,num_epochs=32)
Am i right, correct me if i am wrong?