我用的是你去年的版本，今年重新装了一个pytorch 训练densenet的时候正常 但是测试的时候卡在第一个batchsize 不动了 其他模型都没问题

[ZezeTao]

Describe the bug
A clear and concise description of what the bug is.

To Reproduce
Steps to reproduce the behavior:

1. Go to '...'
2. Click on '....'
3. Scroll down to '....'
4. See error

Expected behavior
A clear and concise description of what you expected to happen.

Screenshots
If applicable, add screenshots to help explain your problem.

Desktop (please complete the following information):

• OS: [e.g. iOS]
• Browser [e.g. chrome, safari]
• Version [e.g. 22]

Smartphone (please complete the following information):

• Device: [e.g. iPhone6]
• OS: [e.g. iOS8.1]
• Browser [e.g. stock browser, safari]
• Version [e.g. 22]

Additional context
Add any other context about the problem here.

[layumi]

@ZezeTao
你好 你用的是windows还是ubuntu？
可能是由于torch.compile 加速导致的。 可以试试 把对应的行 注释掉。

另外，densenet 再等一等 能跑起来么？