Support torch.compile

[skingorz]

🚀 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

[github-actions]

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.

[github-actions]

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