Comments (2)
I'm not sure about that. We refer to the code from the PredRNN repository, but it doesn't look like np.array
is necessary here.
from openstl.
The reason for slowness is that the above function is called in the forward pass of ConvLSTM, and since numpy operations happen over CPU and they then have to be converted to torch tensors and copied to GPU, therefore making the overall forward pass quite slow. Changing the above function to use torch tensors and GPU should fix the slowness.
from openstl.
Related Issues (20)
- SwinLSTM_D method HOT 2
- mask_true arguments HOT 3
- training log HOT 5
- LOSS smaller that mse HOT 1
- setting of mask_true argument HOT 2
- Lightening Version Enhancement: Add Tensorboard monitoring and train from resume functions HOT 1
- export onnx HOT 2
- Are tests still maintained? HOT 2
- for more detailed tutorials HOT 2
- apply for taxibj dataset HOT 1
- 关于Kitti数据集序列数目问题 HOT 2
- Fix a bug for onecycle scheduler usage in Pytorch Lightening Version HOT 2
- where can I find implement of TAU HOT 3
- lighting版本如果用cpu跑,需要在api/exp.py的trainer_config参数进行指定?
- dataset_constant中weather_mv_4_4_s6_5_625的idx_in、idx_out疑问 HOT 1
- For Lightning Version, train.sh/test.sh is not provided. HOT 3
- 如何选择hid_S,hid_T,N_S,N_T的大小 HOT 2
- Data standardization for Human3.6M HOT 2
- 如何指定单个GPU进行训练 HOT 2
- 使用提供的config训练2000轮达不到benchmark上的指标 HOT 17
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 openstl.