search

chenyaofo / CTNAS

[CVPR 2021] Contrastive Neural Architecture Search with Neural Architecture Comparators
BSD 3-Clause "New" or "Revised" License
39 stars 5 forks source link

"pseudo_set" for train() function #4

Closed nutellamok closed 3 years ago

nutellamok commented 3 years ago

Where do you define the "pseudo_set" for the train() function in line 53 of train.py? Right now, it appears that you are feeding "None" as the pseudo-set variable: accuracy, loss = train(epoch=epoch, labeled_loader=train_loader, pseudo_set=None, pseudo_ratio=args.pseudo_ratio, nac=nac, criterion=criterion, optimizer=nac_optimizer)

nutellamok commented 3 years ago

Nvmd, I found it! Thank you for releasing your code :)