search

Chiaraplizz / ST-TR

Spatial Temporal Transformer Network for Skeleton-Based Activity Recognition
MIT License
299 stars 58 forks source link

code can't run #29

Closed 1274739295 closed 3 years ago

1274739295 commented 3 years ago

When I run the code, no matter which of the three data sets I use, I get to epoch with the following problems. Using only the node and using the joint data of the node and the skeleton results in an error at 10 epoch. So I wonder if the code is storing other unimportant data during execution, please kindly comment.

RuntimeError:CUDA out of memory.Tried to allocate 29.38 MiB (GPU 0; 7.79 GiB total capacity; 6.61 GiB already allocated;21.38 MiB free; 91.98 MiB cached)

Chiaraplizz commented 3 years ago

Hi,

Did you try to reduce the batch size?

Chiara