Cpt.Captain/CREStereo-pytorch-nxt
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
66127ec2aa
CREStereo-pytorch-nxt/nets/attention
History
Ibai 3396d2fd06 Fix CUDA inference
2022-04-08 12:00:32 +09:00
..
__init__.py Initial commit 2022-04-08 00:18:27 +09:00
linear_attention.py Initial commit 2022-04-08 00:18:27 +09:00
position_encoding.py Fix CUDA inference 2022-04-08 12:00:32 +09:00
transformer.py Initial commit 2022-04-08 00:18:27 +09:00