Pretrained weights

[abhigoku10]

@Jarr0d hi thanks for opesourcing the code based , can you please share ur pretrained weights on google drive or one drive

Thanks in adavance

[Jarr0d]

Hi,
We are sorry that the model weight is not available. You can train your own model following the README, it will only take about a few hours.
We did not use any training skills so the results should be easily re-produced.

[abhigoku10]

@Jarr0d thanks for the response , i tried to train the model with the source code on V100 machine with 16gb memory but after 6/8 epochs uncertain most times, the terminal kills itself without any error ?
I tried changing the batchsize from 64 to 4 and num_workers from 3 to 1 even after this i am getting error . Can you please share your thoughts