Comments (3)
Hi,
After setting worker = 0, I got new error:
"ValueError: Expected more than 1 value per channel when training, got input size [1, 512, 1, 1]"
Traceback (most recent call last):
File "", line 1, in
runfile('C:/Users/jakaria/DEXTR-PyTorch/train_pascal.py', wdir='C:/Users/jakaria/DEXTR-PyTorch')
File "C:\Users\jakaria\AppData\Local\conda\conda\envs\tensorflow\lib\site-packages\spyder\utils\site\sitecustomize.py", line 705, in runfile
execfile(filename, namespace)
File "C:\Users\jakaria\AppData\Local\conda\conda\envs\tensorflow\lib\site-packages\spyder\utils\site\sitecustomize.py", line 102, in execfile
exec(compile(f.read(), filename, 'exec'), namespace)
File "C:/Users/jakaria/DEXTR-PyTorch/train_pascal.py", line 148, in
output = net.forward(inputs)
File "C:\Users\jakaria\DEXTR-PyTorch\networks\deeplab_resnet.py", line 196, in forward
x = self.layer5(x)
File "C:\Users\jakaria\AppData\Local\conda\conda\envs\tensorflow\lib\site-packages\torch\nn\modules\module.py", line 357, in call
result = self.forward(*input, **kwargs)
File "C:\Users\jakaria\DEXTR-PyTorch\networks\deeplab_resnet.py", line 118, in forward
priors = [F.upsample(input=stage(feats), size=(h, w), mode='bilinear') for stage in self.stages]
File "C:\Users\jakaria\DEXTR-PyTorch\networks\deeplab_resnet.py", line 118, in
priors = [F.upsample(input=stage(feats), size=(h, w), mode='bilinear') for stage in self.stages]
File "C:\Users\jakaria\AppData\Local\conda\conda\envs\tensorflow\lib\site-packages\torch\nn\modules\module.py", line 357, in call
result = self.forward(*input, **kwargs)
File "C:\Users\jakaria\AppData\Local\conda\conda\envs\tensorflow\lib\site-packages\torch\nn\modules\container.py", line 67, in forward
input = module(input)
File "C:\Users\jakaria\AppData\Local\conda\conda\envs\tensorflow\lib\site-packages\torch\nn\modules\module.py", line 357, in call
result = self.forward(*input, **kwargs)
File "C:\Users\jakaria\AppData\Local\conda\conda\envs\tensorflow\lib\site-packages\torch\nn\modules\batchnorm.py", line 37, in forward
self.training, self.momentum, self.eps)
File "C:\Users\jakaria\AppData\Local\conda\conda\envs\tensorflow\lib\site-packages\torch\nn\functional.py", line 1011, in batch_norm
raise ValueError('Expected more than 1 value per channel when training, got input size {}'.format(size))
ValueError: Expected more than 1 value per channel when training, got input size [1, 512, 1, 1]
from dextr-pytorch.
I have 4 gb nvidia gtx 1050 GPU. Is it ok to train with that? can I reduce batch size to 1?
from dextr-pytorch.
Hello, you cannot use batch size 1 when training (refer to this issue). Therefore, batch size has to be at least 2. Morevoer, if len(train_set) % batch_size =1 you will encounter the same error. Just write in line 142 of train_pascal.py
:
if inputs.shape[0] == 1:
continue
Let us know if that fixes your problem.
from dextr-pytorch.
Related Issues (20)
- MS Coco Training Code HOT 1
- UR CODE WAS COPIED FOR SELL NOW HOT 1
- PASCAL Context Background (Stuff) Classes dataloader
- Results of dextr_pascal and dextr_coco models HOT 2
- Training time of PASCAL VOC 2012 HOT 1
- What if generated mask is not correct? HOT 1
- I can't click the extreme points in the picture HOT 1
- IndexError: index 0 is out of bounds for axis 0 with size 0 HOT 3
- RuntimeError: ONNX export failed: Couldn't export operator aten::adaptive_avg_pool2d
- Issue with running the demo file HOT 2
- Results on background classes of PASCAL Context HOT 1
- OSError: [WinError 126] The specified module could not be found HOT 1
- Speed metrics ? HOT 2
- How to train this model on other dataset? HOT 1
- Performance of DEXTR with 2 or 3 points HOT 3
- process about clicks more than 4 points HOT 1
- demo.py blocks at line 49
- Question about Loss HOT 3
- List out of range HOT 2
- Train on Custom dataset in COCO style annotation
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from dextr-pytorch.