microsoft/Semi-supervised-learning

Support torch.compile

skingorz opened this issue ยท 2 comments

๐Ÿš€ Feature

I hope the repo support torch.compile.

Motivation

As a key feature of Pytorch 2.0, I hope USB can also support torch.compile.

Additional context

I have try to add the following code (line 390-392) to train.py. But when the second epoch ends and evals the model ,

eval_dict = algorithm.evaluate('eval')

it always occurs an assert error. My command is

python train.py --c config/usb_cv/freematch/freematch_cifar100_400_0.yaml
image

This issue is stale because it has been open 30 days with no activity. Remove stale label or comment or this will be closed in 5 days.

This issue was closed because it has been stalled for 5 days with no activity.