densenet.pytorch
densenet.pytorch copied to clipboard
bamos
A PyTorch implementation of DenseNet.
https://github.com/bamos/densenet.pytorch/blob/d1cd5e1957975628286e516512c6d1c14430f810/densenet.py#L113-L114 I solve the issue by changing the lines of codes to `out = self.dense3(out) out = self.relu(self.bn1(out)) out = F.avg_pool2d(out, 8) out = out.view(-1, self.nChannels)` where `self.relu` has been...
Hi, I modified your code to train a model with my own dataset, and I am trying to load the model saved as "latest.pth" to do some tests. However, I...
when I fine-tuning my dataset(96*96*3),error : RuntimeError: size mismatch, m1: [2394 x 7], m2: [16758 x 2] at /opt/conda/conda-bld/pytorch_1532502421238/work/aten/src/TH/generic/THTensorMath.cpp:2070
I have slightly modified your algorithm and have adapted it for two classes (k = 12, reduction = 0.5, bottleneck = True). When I train it on Cat and Dog...
Hi @bamos, According to the paper, the number of channels is twice the grow_rate only for the BC type DenseNets. Since you are giving room to not have bottleneck, we...
I used the default setting(which I think is Densenet-12-BC with data augmentation) on cifar-100(via just changing the name of `dataset `class and the `nClasses `variable). The training curve looks like...
Hi, I found a problem in [compute-cifar10-mean.py#L31](https://github.com/bamos/densenet.pytorch/blob/master/compute-cifar10-mean.py#L31). The shape of `torchvision.datasets.CIFAR10.train_data` should be (50000, 32, 32, 3) instead of (50000, 3, 32, 32), so the code in line#31 should be...
Hi author Thanks for sharing your code. I notice in README you said "Multi-gpu help wanted". If you are indicating data parallelism, then it can be implemented in several lines...
Hey Brandon, Do you have an implementation of FCN DenseNet https://arxiv.org/pdf/1611.09326.pdf paper? Thanks!
