Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Model Writing Issue with txt Format. #10

Open
Gaaaavin opened this issue Jul 25, 2024 · 0 comments
Open

Model Writing Issue with txt Format. #10

Gaaaavin opened this issue Jul 25, 2024 · 0 comments

Comments

@Gaaaavin
Copy link

Gaaaavin commented Jul 25, 2024

When running preprocess/auto_reorient.py with --model_type txt flag, the saved txt file will contain tensors.

After some investigation, I find out this is due to this line in the file.

In this line, the pt.xyz is a torch tensor due to previous pytorch operations. Directly applying the * operator would result in three individul tensors, which won't be written properly as numbers to the text file.

This can be simply solved by first converting the tensor into an numpy array, i.e. *pt.xyz.numpy().

I have opened a pull request #11 for this issue.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant