gaozhihan / gaozhihan/PreDiff

"RuntimeError: Tensors must be CUDA and dense" when i run train_vae_sevirlr.py for train

Open
#24 3 comments 0 reactions 1 assignee Claimed by @gaozhihan View on GitHub
bug
Dominant language
Python
Stars
159
Forks
10
PR merge metrics
No merged PRs in 30d

Description

It seems that there is an error when using torchmetrics to calculate the loss indicator. When I loaded the pre-trained model weights for testing, and subsequent training prediff or test prediff, no such error was reported. I was wondering if anyone has encountered this problem? I hope you can take the time to answer my questions
![微信截图_20240315194140](https://github.com/gaozhihan/PreDiff/assets/129286270/fe064ab3-28b9-4c23-8f6b-44909e0937ee)

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.