Comments (4)
Mosquitoes
I have the same problem, mine is 1080ti, I have changed num_pts and chunk_size to 1, but still not working. Can the author publish a small data set that can fit 12GB of video memory?
from omnimotion.
Sorry for the confusion, num of points and chunk size will only control the memory usage for training but not preprocessing. Could you provide more details about which part of the processing code leads to the OOM error? One likely reason is lines like this consume too much memory.
from omnimotion.
ah I see. Thank you,
So yes I am able compute the Dino Features but the moment it begins doing the pairwise optical flows it instantly crashes. It needs a very large amount of memory. Can we scale this down or modify it as well in pre? Thank you for any advice you may be able to provide!
from omnimotion.
I am trying to train on some videos of Mosquitoes and am doing some preprocessing. I am running into
torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 12.00 GiB total capacity; 9.91 GiB already allocated; 0 bytes free; 11.27 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
I am on a 3080ti. Inside of config.py i reduced the default num of points to 20 and the chunk size to a measly 100. Yet still memory errors. Any suggestions? Ran nvidia-smi and have nothing else hogging Gpu. trying to squeeze down can't afford an A100!
I change num_pairs to 4, it runs successfully on 3060 with 12GB memory.
from omnimotion.
Related Issues (20)
- preparation of custom data set
- Training and evaluating model on TAP-Vid DAVIS produces different results HOT 10
- How to accelerate training speed? HOT 1
- Reporting mistakes during training HOT 2
- A question about blending weight. HOT 9
- The TAPNet loader Module
- The given checkpoint do not match all the model, and it's hard to reproduce the result HOT 6
- Question about the depth consistency loss HOT 2
- Hello! This is a question about how to perform online operations after training is complete. HOT 3
- Train all frames or sample some? HOT 2
- Particle Tracking Results HOT 3
- Does it have to be trained and optimized for every new video? HOT 1
- the frame resolution when evaluating on TAP-Vid HOT 1
- Will the model weights and testing code be open-sourced HOT 1
- Transformation matrix
- Evaluate the trained checkpoints and the provided checkpoints, and the results of the metrics are inconsistent
- What may be the reason for not generating visual trajectories.
- Can I transform the input of INN into three-dimensional coordinates with an additional fixed fourth dimension?
- track fast moving objects
- Could you share the code for RAFT-C and RAFT-D evaluation in table 1? Thank you! HOT 1
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 omnimotion.