Open cardoso-neto opened 5 years ago
Created an env file which you can use. Just download the yaml from my repo https://github.com/cardoso-neto/ssds.pytorch/blob/5384d8c27de82919eab956655a88834e29ac8c2d/ssds.yaml and run conda env create -f ssds.yaml
conda env create -f ssds.yaml
This will, however, require one small modification to line 45 of ssds_train.py from:
ssds_train.py
self.priors = self.priorbox.forward()
to:
self.priors = Variable(self.priorbox.forward())
The error message is quite clear and straightforward, suggesting to cast the torch.Tensor to an autograd.Variable.
torch.Tensor
autograd.Variable
After fiddling a bit with different Python/Pytorch versions, this is seemed to work: Python=3.6 Pytorch=0.3.1
Not really an issue. Just wanted to leave this somewhere where it'd be helpful.
Created an env file which you can use. Just download the yaml from my repo https://github.com/cardoso-neto/ssds.pytorch/blob/5384d8c27de82919eab956655a88834e29ac8c2d/ssds.yaml and run
conda env create -f ssds.yamlThis will, however, require one small modification to line 45 of
ssds_train.pyfrom:to:
The error message is quite clear and straightforward, suggesting to cast the
torch.Tensorto anautograd.Variable.After fiddling a bit with different Python/Pytorch versions, this is seemed to work: Python=3.6 Pytorch=0.3.1
Not really an issue. Just wanted to leave this somewhere where it'd be helpful.