Code changes for custom classes

weixmath / view-GCN

Pytorch code for view-GCN (CVPR2020)

77 stars 17 forks source link

Code changes for custom classes #14

Open piseabhijeet opened 3 years ago

piseabhijeet commented 3 years ago

Hi @weixmath

What changes are required in the code to make it train on 5 classes? I tried modifying the code, it has high train (100%) and validation accuracy (95%) but doesn't work on test dataset - 30% accuracy

Thanks in advance.