search

justusschock / deep_alignment_network_pytorch

PyTorch Implementation of the Deep Alignment Network
https://justusschock.github.io/deep_alignment_network_pytorch/
BSD 2-Clause "Simplified" License
38 stars 8 forks source link

Hi, How about the performance in different datasets? this version can reach the the theano version? Look forward for your kind help! #1

Closed wqz960 closed 5 years ago

wqz960 commented 5 years ago

Can you provide me the pretrained model?

justusschock commented 5 years ago

Unfortunately I cannot provide you a pretrained model, and I haven't even trained a model on a freely-available dataset (the HELEN notebook is just to show the use), so I can't confirm that either (sorry!).

What I can confirm is, that on an internal dataset it reaches comparable results to the theano version.

wqz960 commented 5 years ago

@justusschock It seems you did not provide the dataset preprocess script, can you provide it?

wqz960 commented 5 years ago

@justusschock use shapedata to make data argument?

justusschock commented 5 years ago

it's all done in the example notebook via shapedata.

The data isn't preprocessed completely before training, but it is done OTF during data loading via shapenet