Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove hardcoded cuda architecture

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-18 20:44:32 +02:00
parent c1d93bd008
commit ef3a62f399
3 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/export_model_weights.py
View File

@@ -25,4 +25,4 @@ def export_model_weights(model: torch.nn.Module, filename):
f.seek(0)
f.write(struct.pack('q', len(header)))
f.write(header.encode('utf-8'))
f.write(header.encode('utf-8'))