andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:sugarme/gotch into 1.9

Browse Source
This commit is contained in:
sugarme 2021-07-23 10:51:04 +10:00
commit 842dd45f52
3 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
CHANGELOG.md
View File

@ -7,6 +7,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
## [Unreleased]
## [Nofix]
- ctype `long` caused compiling error in MacOS as noted on [#44]. Not working on linux box.
## [0.3.14]
- Fixed temporary fix huge number of learning group returned from C at `libtch/tensor.go AtoGetLearningRates`
- Fixed incorrect `nn.AdamWConfig` and some documentation.
- Fixed - reworked on `vision.ResNet` and `vision.DenseNet` to fix incorrect layers and memory leak
@ -18,9 +23,6 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
- [#45] Fixed `init/RandInt` incorrect initialization
- [#48] Fixed `init/RandInit` when init with mean = 0.0.
## [Nofix]
- ctype `long` caused compiling error in MacOS as noted on [#44]. Not working on linux box.
## [0.3.13]
- Fixed multiple memory leakage at `vision/image.go`
- Fixed memory leakage at `dutil/dataloader.go`

6
README.md
View File

@ -51,7 +51,7 @@ Gotch is in active development mode and may have API breaking changes. Feel free
```bash
wget https://raw.githubusercontent.com/sugarme/gotch/master/setup-gotch.sh
chmod +x setup-gotch.sh
export CUDA_VER=cpu && export GOTCH_VER=v0.3.13 && bash setup-gotch.sh
export CUDA_VER=cpu && export GOTCH_VER=v0.3.14 && bash setup-gotch.sh
```
### GPU
@ -89,9 +89,9 @@ Gotch is in active development mode and may have API breaking changes. Feel free
wget https://raw.githubusercontent.com/sugarme/gotch/master/setup-gotch.sh
chmod +x setup-gotch.sh
# CUDA 10.1
export CUDA_VER=10.1 && export GOTCH_VER=v0.3.13 && bash setup-gotch.sh
export CUDA_VER=10.1 && export GOTCH_VER=v0.3.14 && bash setup-gotch.sh
# CUDA 11.0
export CUDA_VER=11.0 && export GOTCH_VER=v0.3.13 && bash setup-gotch.sh
export CUDA_VER=11.0 && export GOTCH_VER=v0.3.14 && bash setup-gotch.sh
```
## Examples

2
setup-gotch.sh
View File

@ -1,6 +1,6 @@
#!/bin/bash
GOTCH_VERSION="${GOTCH_VER:-v0.3.13}"
GOTCH_VERSION="${GOTCH_VER:-v0.3.14}"
CUDA_VERSION="${CUDA_VER:-10.1}"
GOTCH_PATH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION"