Comments (2)
Do you use token label? If not, maybe you should modify "return_dense=True, mix_token=True" to "return_dense=False, mix_token=False" in
Line 462 in 068a583
when i fixed your pointed codes, it appeared this error :
File "main.py", line 652, in main
train_metrics = train_one_epoch(epoch,
File "main.py", line 786, in train_one_epoch
loss = loss_fn(output, target)
File "C:\Program Files\Anaconda3\envs\yolov5-v4.0\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "C:\Program Files\Anaconda3\envs\yolov5-v4.0\lib\site-packages\torch\nn\modules\loss.py", line 961, in forward
return F.cross_entropy(input, target, weight=self.weight,
File "C:\Program Files\Anaconda3\envs\yolov5-v4.0\lib\site-packages\torch\nn\functional.py", line 2468, in cross_entropy
return nll_loss(log_softmax(input, 1), target, weight, None, ignore_index, None, reduction)
File "C:\Program Files\Anaconda3\envs\yolov5-v4.0\lib\site-packages\torch\nn\functional.py", line 2264, in nll_loss
ret = torch._C._nn.nll_loss(input, target.long(), weight, _Reduction.get_enum(reduction), ignore_index) # target
RuntimeError: multi-target not supported at C:/w/b/windows/pytorch/aten/src\THCUNN/generic/ClassNLLCriterion.cu:15
i fixed "loss = loss_fn(output, target)" to "loss = loss_fn(output, target.squeeze())", this error still appears
what can i do?
from volo.
Do you use token label?
If not, maybe you should modify "return_dense=True, mix_token=True" to "return_dense=False, mix_token=False" in
Line 462 in 068a583
from volo.
Related Issues (20)
- Halonets declare a SOTA in CVPR, any contribution to conference?
- Some thoughts about volo
- Colab Notebook doesn't work & gives wrong results. HOT 4
- Finetune with 512 image size HOT 2
- Balanced class weight HOT 1
- Will you release the codes for Semantic Segmentation?
- Could you share the training script for training volo_d5 ↑512?
- run_error HOT 1
- SystemError: returned NULL without setting an error
- May you provide the FLOPs calculation script ?
- size mismatch for pos_embed: copying a param with shape torch.Size([1, 14, 14, 768]) from checkpoint, the shape in current model is torch.Size([1, 14, 14, 512]). HOT 1
- RuntimeError: The size of tensor a (28) must match the size of tensor b (14) at non-singleton dimension 2 HOT 2
- is the order of v_pj(x) result in algoritm1 correct ?
- The code for Semantic Segmentation?
- When training own dataset, an error occurs when changing numberclasses to the corresponding category. If it is the default, it will report an error
- volo-d1 training without token label data HOT 1
- pre trained model file is broken. HOT 1
- Question about computational complexity formulation of Outlooker Attention
- Ablation study and official code problem
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 volo.