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

Hi , can I train this model on my own dataset ? #41

Closed
jediofgever opened this issue Nov 6, 2019 · 5 comments
Closed

Hi , can I train this model on my own dataset ? #41

jediofgever opened this issue Nov 6, 2019 · 5 comments

Comments

@jediofgever
Copy link

No description provided.

@WangYueFt
Copy link
Owner

You probably need to do two things:

  1. implement a new dataloader or modify the existing one.
  2. change the input placeholder to be compatible with your data.

@jediofgever
Copy link
Author

@WangYueFt , Thank you for your reply, Also , if my point cloud contains more than say 2048 points, what will be expected behaviour of network to it , do I need to have strictly 2048 points ?

@WangYueFt
Copy link
Owner

@WangYueFt , Thank you for your reply, Also , if my point cloud contains more than say 2048 points, what will be expected behaviour of network to it , do I need to have strictly 2048 points ?

You don't need to. But in each batch, you have to pad them into the same number of points.

@jediofgever
Copy link
Author

@WangYueFt, I did create data files same as .h5 format and trained network but even in epoch 0 I got 0.99 train accuracy.

Train 0, loss: 1.376507, train acc: 0.993896, train avg acc: 0.993896
Test 0, loss: 1.233287, test acc: 1.000000, test avg acc: 1.000000
Train 1, loss: 1.233263, train acc: 1.000000, train avg acc: 1.000000
Test 1, loss: 1.233124, test acc: 1.000000, test avg acc: 1.000000
Train 2, loss: 1.233195, train acc: 1.000000, train avg acc: 1.000000
Test 2, loss: 1.233121, test acc: 1.000000, test avg acc: 1.000000
Train 3, loss: 1.233166, train acc: 1.000000, train avg acc: 1.000000
Test 3, loss: 1.233120, test acc: 1.000000, test avg acc: 1.000000
Train 4, loss: 1.233150, train acc: 1.000000, train avg acc: 1.000000
Test 4, loss: 1.233116, test acc: 1.000000, test avg acc: 1.000000

Does this mean model if overfitting ?

@AdeV-Oly
Copy link

AdeV-Oly commented Jun 7, 2022

Hey there @jediofgever

I'm attempting to train (the equivariant version (VNN) of) this model on my own dataset as well. I think I may be overlooking something, as I'm running into an error.

If you or @WangYueFt could take a look at it and give me some pointers, or if I need to add some more info, that'd be much appreciated.

Link to my issue on the VNN github: FlyingGiraffe/vnn#13

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

3 participants