davrot/pytorch-sbs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update README.md

Browse source
This commit is contained in:
David Rotermund 2023-01-15 16:13:55 +01:00 committed by GitHub
parent 08550d1909
commit ac85c41302
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
README.md
View file

@ -26,3 +26,15 @@ In addition your Python installation needs the PyBind11 package installed. You m
pip install pybind11
The Makefile uses clang as a compiler. If you want something else then you need to change the Makefile.
For CUDA I used version 12.0.
# C++
Three .json config files are required:
dataset.json : Information about the dataset
network.json : Describes the network architecture
def.json : Controlls the other parameters
If you want to load existing weights, just put them in a sub-folder called Previous