mirror of https://github.com/JDAI-CV/DCL.git
parent
1b523451aa
commit
5e529402b0
1
test.py
1
test.py
|
@ -82,7 +82,6 @@ if __name__ == '__main__':
|
||||||
|
|
||||||
setattr(dataloader, 'total_item_len', len(data_set))
|
setattr(dataloader, 'total_item_len', len(data_set))
|
||||||
|
|
||||||
save_result = Submit_result(args.dataset)
|
|
||||||
cudnn.benchmark = True
|
cudnn.benchmark = True
|
||||||
|
|
||||||
model = MainModel(Config)
|
model = MainModel(Config)
|
||||||
|
|
Loading…
Reference in New Issue