SupContrast
SupContrast copied to clipboard
May be a bug?
Shell line 292 in main_supcon.py
save_model(model, optimizer, opt, opt.epochs, save_file)
be
save_model(model, optimizer, opt, epoch, save_file)
?
Because there are no need to save both opt and opt.epochs. Maybe epoch (the epoch now) is more important?
"Shall line...", not "Shell". Sorry.