Kunguma gokul S

When I am trying to do Pretraining for custom dataset I am getting this error ,In this i am just using two images and two captions

1

Start training Traceback (most recent call last): File "pretrain.py", line 173, in main(args, config) File "pretrain.py", line 131, in main train_stats = train(model, data_loader, optimizer, epoch, device, config) File "pretrain.py",...

Start training Traceback (most recent call last): File "pretrain.py", line 173, in main(args, config) File "pretrain.py", line 131, in main train_stats = train(model, data_loader, optimizer, epoch, device, config) File "pretrain.py",...

Custom dataset

2

1) How can we do it for Custom dataset and how the labels should be? 2)Whether can we able to generate samples for Specific class...Suppose I want to generate for...