Closed noelcodes closed 6 years ago
Sorry, I cannot provide the dataset due to the morph2 dataset requires an application and some payment. I have no idea why the dataset website is down. You have to ask the authors of the dataset.
Since morph db is dead, therefore i am trying to modify run_ssrnet.sh. I noticed u have nested for looped imdb + wiki with i,j. I am confused, because your demo is only taking either wiki OR imdb weights. I planned to cut down train time by not looping, but not sure if this mess up the model concept.
So I want to clarify, your purpose of the for-loop, is it because you are taking the last train weights from imdb for start weights for wiki, then continued loop using last wiki weight for start imdb and so on and on...? If this is so, then for loop is mandatory.
I was planning to train only this: but not sure. python SSRNET_train.py --input ../data/imdb_db.npz --db imdb --netType1 1 --netType2 1
No. The for loop is just for different hyper parameters. Different model from different hyper parameters do not take each others pretrained weight.
The paper uses --netType1 4 --netType2 4.
Any idea why? It throw me an error when I run training without loop, just this cmd> python SSRNET_train.py --input ../data/imdb_db.npz --db imdb --netType1 1 --netType2 1 Error: UnicodeDecodeError: 'rawunicodeescape' codec can't decode bytes in position 838-839: truncated \uXXXX
==================================================================================================== Total params: 40,915 Trainable params: 40,531 Non-trainable params: 384
DEBUG:root:Saving model...
Traceback (most recent call last):
File "SSRNET_train.py", line 136, in
Morph2 dataset website is dead. Do you happen to have the dataset for download?