diff --git a/README.md b/README.md index f679d3a..d24a4e9 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,6 @@ +# Fork of gotch modified to work with my fyp project + + # Gotch [![License](https://img.shields.io/:license-apache-blue.svg)](https://opensource.org/licenses/Apache-2.0)[![Go.Dev reference](https://img.shields.io/badge/go.dev-reference-007d9c?logo=go&logoColor=white&style=flat-square)](https://pkg.go.dev/github.com/sugarme/gotch?tab=doc) diff --git a/device.go b/device.go index 38df73e..899e48f 100644 --- a/device.go +++ b/device.go @@ -3,7 +3,7 @@ package gotch import ( "log" - lib "github.com/sugarme/gotch/libtch" + lib "git.andr3h3nriqu3s.com/andr3/gotch/libtch" ) type Device struct { diff --git a/dutil/dataloader.go b/dutil/dataloader.go index 1f76877..b9725a7 100644 --- a/dutil/dataloader.go +++ b/dutil/dataloader.go @@ -4,7 +4,7 @@ import ( "fmt" "reflect" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // DataLoader combines a dataset and a sampler and provides diff --git a/dutil/dataloader_test.go b/dutil/dataloader_test.go index 977d24d..9b509fc 100644 --- a/dutil/dataloader_test.go +++ b/dutil/dataloader_test.go @@ -5,7 +5,7 @@ import ( "reflect" "testing" - "github.com/sugarme/gotch/dutil" + "git.andr3h3nriqu3s.com/andr3/gotch/dutil" ) func TestNewDataLoader(t *testing.T) { diff --git a/dutil/dataset_test.go b/dutil/dataset_test.go index bfc1d96..7d3ab1c 100644 --- a/dutil/dataset_test.go +++ b/dutil/dataset_test.go @@ -4,7 +4,7 @@ import ( "reflect" "testing" - "github.com/sugarme/gotch/dutil" + "git.andr3h3nriqu3s.com/andr3/gotch/dutil" ) func TestNewSliceDataset(t *testing.T) { diff --git a/dutil/kfold_test.go b/dutil/kfold_test.go index 34886f0..7c50aff 100644 --- a/dutil/kfold_test.go +++ b/dutil/kfold_test.go @@ -3,7 +3,7 @@ package dutil_test import ( "testing" - "github.com/sugarme/gotch/dutil" + "git.andr3h3nriqu3s.com/andr3/gotch/dutil" ) func TestNewKFold(t *testing.T) { diff --git a/dutil/sampler_test.go b/dutil/sampler_test.go index 3fbd3e3..e5f5ed7 100644 --- a/dutil/sampler_test.go +++ b/dutil/sampler_test.go @@ -5,7 +5,7 @@ import ( "reflect" "testing" - "github.com/sugarme/gotch/dutil" + "git.andr3h3nriqu3s.com/andr3/gotch/dutil" ) func TestSequentialSampler(t *testing.T) { diff --git a/example/augmentation/main.go b/example/augmentation/main.go index 440b715..0063e03 100644 --- a/example/augmentation/main.go +++ b/example/augmentation/main.go @@ -4,9 +4,9 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/vision" - "github.com/sugarme/gotch/vision/aug" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch/vision/aug" ) func main() { diff --git a/example/basic/main.go b/example/basic/main.go index 3fb336d..285a285 100644 --- a/example/basic/main.go +++ b/example/basic/main.go @@ -3,8 +3,8 @@ package main import ( "fmt" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func main() { diff --git a/example/char-rnn/main.go b/example/char-rnn/main.go index 19dfcc7..bac1fc0 100644 --- a/example/char-rnn/main.go +++ b/example/char-rnn/main.go @@ -4,9 +4,9 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) const ( diff --git a/example/cifar/main.go b/example/cifar/main.go index 09179ca..158cdbf 100644 --- a/example/cifar/main.go +++ b/example/cifar/main.go @@ -12,10 +12,10 @@ import ( "log" "time" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) func convBn(p *nn.Path, cIn, cOut int64) *nn.SequentialT { diff --git a/example/convert-model/main.go b/example/convert-model/main.go index b3a7389..8d018f6 100644 --- a/example/convert-model/main.go +++ b/example/convert-model/main.go @@ -4,7 +4,7 @@ import ( "fmt" "log" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func main() { diff --git a/example/debug-memory/main.go b/example/debug-memory/main.go index ba68e4d..b30d783 100644 --- a/example/debug-memory/main.go +++ b/example/debug-memory/main.go @@ -5,8 +5,8 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) var device string diff --git a/example/jit-train/main.go b/example/jit-train/main.go index a3b6751..64cdfcb 100644 --- a/example/jit-train/main.go +++ b/example/jit-train/main.go @@ -5,10 +5,10 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) var ( diff --git a/example/jit/main.go b/example/jit/main.go index 56e6809..51a93e7 100644 --- a/example/jit/main.go +++ b/example/jit/main.go @@ -9,9 +9,9 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) var ( diff --git a/example/mem/main.go b/example/mem/main.go index d6b39b1..e686cd0 100644 --- a/example/mem/main.go +++ b/example/mem/main.go @@ -4,9 +4,9 @@ import ( "fmt" "math/rand" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) const ( diff --git a/example/mnist-fp16/main.go b/example/mnist-fp16/main.go index d03f6af..2689183 100644 --- a/example/mnist-fp16/main.go +++ b/example/mnist-fp16/main.go @@ -7,10 +7,10 @@ import ( "sync" "time" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) func main() { diff --git a/example/mnist/cnn.go b/example/mnist/cnn.go index 5a0d505..bf9e05f 100644 --- a/example/mnist/cnn.go +++ b/example/mnist/cnn.go @@ -7,10 +7,10 @@ import ( "sync" "time" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) const ( diff --git a/example/mnist/linear.go b/example/mnist/linear.go index 0c10f88..4c3051d 100644 --- a/example/mnist/linear.go +++ b/example/mnist/linear.go @@ -3,9 +3,9 @@ package main import ( "fmt" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) const ( diff --git a/example/mnist/main.go b/example/mnist/main.go index e7e76d3..ab2b5fa 100644 --- a/example/mnist/main.go +++ b/example/mnist/main.go @@ -3,7 +3,7 @@ package main import ( "flag" - "github.com/sugarme/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch" ) var ( diff --git a/example/mnist/nn.go b/example/mnist/nn.go index da38775..66bd91d 100644 --- a/example/mnist/nn.go +++ b/example/mnist/nn.go @@ -5,10 +5,10 @@ import ( "log" "runtime" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) const ( diff --git a/example/neural-style-transfer/main.go b/example/neural-style-transfer/main.go index 5429ba8..6b41ff8 100644 --- a/example/neural-style-transfer/main.go +++ b/example/neural-style-transfer/main.go @@ -10,10 +10,10 @@ import ( "path/filepath" "time" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) const ( diff --git a/example/pickle/main.go b/example/pickle/main.go index fa79824..e054d40 100644 --- a/example/pickle/main.go +++ b/example/pickle/main.go @@ -4,8 +4,8 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/pickle" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/pickle" ) func main() { diff --git a/example/pretrained-model/main.go b/example/pretrained-model/main.go index 8e1dd1d..ae4222c 100644 --- a/example/pretrained-model/main.go +++ b/example/pretrained-model/main.go @@ -10,10 +10,10 @@ import ( "path/filepath" "strings" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) var ( diff --git a/example/scheduler/main.go b/example/scheduler/main.go index a283da0..585b12c 100644 --- a/example/scheduler/main.go +++ b/example/scheduler/main.go @@ -4,9 +4,9 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func main() { diff --git a/example/tensor-grad/main.go b/example/tensor-grad/main.go index 70c1f45..32d04fb 100644 --- a/example/tensor-grad/main.go +++ b/example/tensor-grad/main.go @@ -3,7 +3,7 @@ package main import ( "fmt" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func main() { diff --git a/example/tensor-io/main.go b/example/tensor-io/main.go index 647e487..470c8e6 100644 --- a/example/tensor-io/main.go +++ b/example/tensor-io/main.go @@ -3,8 +3,8 @@ package main import ( "fmt" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func main() { diff --git a/example/transfer-learning/main.go b/example/transfer-learning/main.go index 864fdcf..19bc25d 100644 --- a/example/transfer-learning/main.go +++ b/example/transfer-learning/main.go @@ -9,10 +9,10 @@ import ( "log" "path/filepath" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) var ( diff --git a/example/translation/main.go b/example/translation/main.go index 069be37..5d43a99 100644 --- a/example/translation/main.go +++ b/example/translation/main.go @@ -17,9 +17,9 @@ import ( "log" "math/rand" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) var ( diff --git a/example/yolo/darknet.go b/example/yolo/darknet.go index 99ce46e..0ffd874 100644 --- a/example/yolo/darknet.go +++ b/example/yolo/darknet.go @@ -9,9 +9,9 @@ import ( "strconv" "strings" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) type Block struct { diff --git a/example/yolo/draw.go b/example/yolo/draw.go index a1827c6..0f26e0f 100644 --- a/example/yolo/draw.go +++ b/example/yolo/draw.go @@ -15,8 +15,8 @@ import ( "golang.org/x/image/draw" "golang.org/x/image/font" - "github.com/sugarme/gotch/example/yolo/freetype" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/example/yolo/freetype" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) var ( diff --git a/example/yolo/main.go b/example/yolo/main.go index 307e636..ecbfde4 100644 --- a/example/yolo/main.go +++ b/example/yolo/main.go @@ -8,10 +8,10 @@ import ( "path/filepath" "sort" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) const ( diff --git a/go.mod b/go.mod index 6248e03..cb65194 100644 --- a/go.mod +++ b/go.mod @@ -1,4 +1,4 @@ -module github.com/sugarme/gotch +module git.andr3h3nriqu3s.com/andr3/gotch go 1.20 diff --git a/half/float16_bench_test.go b/half/float16_bench_test.go index c1ed12a..e5d9b19 100644 --- a/half/float16_bench_test.go +++ b/half/float16_bench_test.go @@ -6,7 +6,7 @@ import ( "math" "testing" - float16 "github.com/sugarme/gotch/half" + float16 "git.andr3h3nriqu3s.com/andr3/gotch/half" ) // prevent compiler optimizing out code by assigning to these diff --git a/half/float16_test.go b/half/float16_test.go index 073c267..230a69d 100644 --- a/half/float16_test.go +++ b/half/float16_test.go @@ -11,7 +11,7 @@ import ( "math" "testing" - float16 "github.com/sugarme/gotch/half" + float16 "git.andr3h3nriqu3s.com/andr3/gotch/half" ) // wantF32toF16bits is a tiny subset of expected values diff --git a/nn/batch-norm.go b/nn/batch-norm.go index 46b66b9..8327ed2 100644 --- a/nn/batch-norm.go +++ b/nn/batch-norm.go @@ -5,7 +5,7 @@ package nn import ( "log" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Batch-normalization config. diff --git a/nn/conv-transpose.go b/nn/conv-transpose.go index 4e1b146..ddef85f 100644 --- a/nn/conv-transpose.go +++ b/nn/conv-transpose.go @@ -5,7 +5,7 @@ package nn import ( "log" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) type ConvTranspose1DConfig struct { diff --git a/nn/conv.go b/nn/conv.go index baeef66..a598cce 100644 --- a/nn/conv.go +++ b/nn/conv.go @@ -7,7 +7,7 @@ import ( "math" "reflect" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Conv1DConfig: diff --git a/nn/func.go b/nn/func.go index 727a531..eb4f34f 100644 --- a/nn/func.go +++ b/nn/func.go @@ -3,7 +3,7 @@ package nn // Layers defined by closure import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) type Func struct { diff --git a/nn/init.go b/nn/init.go index b94a180..4c4d5d3 100644 --- a/nn/init.go +++ b/nn/init.go @@ -6,8 +6,8 @@ import ( "math" "strings" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) type Init interface { diff --git a/nn/init_test.go b/nn/init_test.go index 88cf3e5..826eb26 100644 --- a/nn/init_test.go +++ b/nn/init_test.go @@ -5,8 +5,8 @@ import ( "testing" "time" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Test whether InitTensor() can cause memory blow-up due to accumulate gradient. diff --git a/nn/jit.go b/nn/jit.go index 91fa1e3..0b8a427 100644 --- a/nn/jit.go +++ b/nn/jit.go @@ -5,7 +5,7 @@ import ( "log" "strings" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // TrainableCModule is a trainable version of JIT Pytorch module diff --git a/nn/layer-norm.go b/nn/layer-norm.go index 3d68e1a..4ce4483 100644 --- a/nn/layer-norm.go +++ b/nn/layer-norm.go @@ -2,7 +2,7 @@ package nn // A layer-normalization layer. import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Layer-normalization config. diff --git a/nn/linear.go b/nn/linear.go index 6216a23..78e128a 100644 --- a/nn/linear.go +++ b/nn/linear.go @@ -6,7 +6,7 @@ import ( "fmt" "math" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // LinearConfig is a configuration for a linear layer diff --git a/nn/loss.go b/nn/loss.go index 397dfb7..fb156ce 100644 --- a/nn/loss.go +++ b/nn/loss.go @@ -1,7 +1,7 @@ package nn import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) type lossFnOptions struct { diff --git a/nn/optimizer.go b/nn/optimizer.go index e882e86..bc368ba 100644 --- a/nn/optimizer.go +++ b/nn/optimizer.go @@ -7,7 +7,7 @@ import ( "log" "math" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Optimizer is a struct object to run gradient descent. diff --git a/nn/optimizer_test.go b/nn/optimizer_test.go index 2d44d17..7642754 100644 --- a/nn/optimizer_test.go +++ b/nn/optimizer_test.go @@ -4,9 +4,9 @@ import ( "fmt" "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func TestOptimizer(t *testing.T) { diff --git a/nn/other.go b/nn/other.go index e604cbc..c2d779c 100644 --- a/nn/other.go +++ b/nn/other.go @@ -1,7 +1,7 @@ package nn import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Dropout: diff --git a/nn/rnn.go b/nn/rnn.go index b7d02bf..fd46fdc 100644 --- a/nn/rnn.go +++ b/nn/rnn.go @@ -3,8 +3,8 @@ package nn import ( "fmt" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) type State interface{} diff --git a/nn/rnn_test.go b/nn/rnn_test.go index bd87cfd..3d5a69c 100644 --- a/nn/rnn_test.go +++ b/nn/rnn_test.go @@ -5,9 +5,9 @@ import ( "reflect" "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func gruTest(rnnConfig *nn.RNNConfig, t *testing.T) { diff --git a/nn/scheduler_test.go b/nn/scheduler_test.go index 376ac2f..ace7d92 100644 --- a/nn/scheduler_test.go +++ b/nn/scheduler_test.go @@ -6,8 +6,8 @@ import ( "math" "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" ) func TestLambdaLR(t *testing.T) { diff --git a/nn/sequential.go b/nn/sequential.go index 3509691..6c057f6 100644 --- a/nn/sequential.go +++ b/nn/sequential.go @@ -3,8 +3,8 @@ package nn // A sequential layer used to chain multiple layers and closures. import ( - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Sequential is a layer (container) that combines multiple other layers. diff --git a/nn/sparse.go b/nn/sparse.go index 3f62175..3a4b0a6 100644 --- a/nn/sparse.go +++ b/nn/sparse.go @@ -3,7 +3,7 @@ package nn // Sparse layers import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Configuration option for an embedding layer. diff --git a/nn/sparse_test.go b/nn/sparse_test.go index 338cb70..0fb333b 100644 --- a/nn/sparse_test.go +++ b/nn/sparse_test.go @@ -4,9 +4,9 @@ import ( "reflect" "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func embeddingTest(embeddingConfig *nn.EmbeddingConfig, t *testing.T) { diff --git a/nn/varstore.go b/nn/varstore.go index 6dae83f..1d6f58c 100644 --- a/nn/varstore.go +++ b/nn/varstore.go @@ -8,8 +8,8 @@ import ( "strings" "sync" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // SEP is a separator to separate path elements in the tensor names. diff --git a/nn/varstore_test.go b/nn/varstore_test.go index c862e68..6392a29 100644 --- a/nn/varstore_test.go +++ b/nn/varstore_test.go @@ -8,9 +8,9 @@ import ( "testing" "time" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func TestVarStoreEntry(t *testing.T) { diff --git a/pickle/pickle_example_test.go b/pickle/pickle_example_test.go index 8a34868..f9de0ca 100644 --- a/pickle/pickle_example_test.go +++ b/pickle/pickle_example_test.go @@ -4,8 +4,8 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/pickle" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/pickle" ) func ExampleLoadInfo() { diff --git a/pickle/serialization.go b/pickle/serialization.go index 2a00484..176586a 100644 --- a/pickle/serialization.go +++ b/pickle/serialization.go @@ -20,9 +20,9 @@ import ( "reflect" "sort" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) const hexMagicNumber = "1950a86a20f9469cfc6c" diff --git a/pickle/storage.go b/pickle/storage.go index 45e35e6..d62c0a1 100644 --- a/pickle/storage.go +++ b/pickle/storage.go @@ -6,8 +6,8 @@ import ( "io" "math" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/half" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/half" ) // This file implements Pytorch storage data types. diff --git a/ts/basic-example_test.go b/ts/basic-example_test.go index 7b6ef77..875864f 100644 --- a/ts/basic-example_test.go +++ b/ts/basic-example_test.go @@ -3,8 +3,8 @@ package ts_test import ( "fmt" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func ExampleTensor_MustArange() { diff --git a/ts/benchmark-conv2d_test.go b/ts/benchmark-conv2d_test.go index d53dfc1..3128b11 100644 --- a/ts/benchmark-conv2d_test.go +++ b/ts/benchmark-conv2d_test.go @@ -3,8 +3,8 @@ package ts_test import ( "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // GOMAXPROCS=8 go test -bench=BenchmarkConv2d -benchtime=100x -run=^a | tee op-conv-bench.txt diff --git a/ts/data.go b/ts/data.go index 4157812..93cba25 100644 --- a/ts/data.go +++ b/ts/data.go @@ -7,7 +7,7 @@ import ( "os" "path/filepath" - "github.com/sugarme/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch" ) // Iter2 is an iterator over a pair of tensors which have the same first dimension diff --git a/ts/data_test.go b/ts/data_test.go index 4177509..f1fadfa 100644 --- a/ts/data_test.go +++ b/ts/data_test.go @@ -8,8 +8,8 @@ import ( "reflect" "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func TestTextData_NewTextData(t *testing.T) { diff --git a/ts/error.go b/ts/error.go index 821383c..3042890 100644 --- a/ts/error.go +++ b/ts/error.go @@ -8,7 +8,7 @@ import ( "runtime/debug" "unsafe" - lib "github.com/sugarme/gotch/libtch" + lib "git.andr3h3nriqu3s.com/andr3/gotch/libtch" ) // ptrToString check C pointer for null. If not null, get value diff --git a/ts/image.go b/ts/image.go index 05d13c0..91db444 100644 --- a/ts/image.go +++ b/ts/image.go @@ -5,7 +5,7 @@ import "C" import ( // "unsafe" - lib "github.com/sugarme/gotch/libtch" + lib "git.andr3h3nriqu3s.com/andr3/gotch/libtch" ) // LoadHwc returns a tensor of shape [height, width, channels] on success. diff --git a/ts/index.go b/ts/index.go index 072cd4a..2b48740 100644 --- a/ts/index.go +++ b/ts/index.go @@ -66,7 +66,7 @@ import ( "log" "reflect" - "github.com/sugarme/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch" ) type NewAxis struct{} diff --git a/ts/index_test.go b/ts/index_test.go index 736d9ae..c3a3c1c 100644 --- a/ts/index_test.go +++ b/ts/index_test.go @@ -4,8 +4,8 @@ import ( "reflect" "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func TestIntegerIndex(t *testing.T) { diff --git a/ts/iter.go b/ts/iter.go index cc81ce3..206c309 100644 --- a/ts/iter.go +++ b/ts/iter.go @@ -4,7 +4,7 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch" ) type Iterator interface { diff --git a/ts/jit.go b/ts/jit.go index d81c014..7079740 100644 --- a/ts/jit.go +++ b/ts/jit.go @@ -14,8 +14,8 @@ import ( "reflect" "unsafe" - "github.com/sugarme/gotch" - lib "github.com/sugarme/gotch/libtch" + "git.andr3h3nriqu3s.com/andr3/gotch" + lib "git.andr3h3nriqu3s.com/andr3/gotch/libtch" ) type CIValue struct { diff --git a/ts/jit_test.go b/ts/jit_test.go index e7ea17a..912bedc 100644 --- a/ts/jit_test.go +++ b/ts/jit_test.go @@ -4,7 +4,7 @@ import ( "reflect" "testing" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func roundTrip(v interface{}, t *testing.T) { diff --git a/ts/must-tensor-generated.go b/ts/must-tensor-generated.go index 79655ab..cc68756 100644 --- a/ts/must-tensor-generated.go +++ b/ts/must-tensor-generated.go @@ -5,7 +5,7 @@ package ts import( "log" - "github.com/sugarme/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch" ) diff --git a/ts/npy.go b/ts/npy.go index 9f8f2e8..f2dba56 100644 --- a/ts/npy.go +++ b/ts/npy.go @@ -11,7 +11,7 @@ import ( "strconv" "strings" - "github.com/sugarme/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch" ) const ( diff --git a/ts/npy_test.go b/ts/npy_test.go index 18ddbdc..80ae17e 100644 --- a/ts/npy_test.go +++ b/ts/npy_test.go @@ -4,8 +4,8 @@ import ( "reflect" "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func TestNpyHeaderParse(t *testing.T) { diff --git a/ts/optimizer.go b/ts/optimizer.go index 2e7e0e8..96bd9f9 100644 --- a/ts/optimizer.go +++ b/ts/optimizer.go @@ -3,7 +3,7 @@ package ts import ( "log" - lib "github.com/sugarme/gotch/libtch" + lib "git.andr3h3nriqu3s.com/andr3/gotch/libtch" ) type COptimizer struct { diff --git a/ts/patch-example_test.go b/ts/patch-example_test.go index 7a956bf..df4da4c 100644 --- a/ts/patch-example_test.go +++ b/ts/patch-example_test.go @@ -3,8 +3,8 @@ package ts_test import ( "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func ExampleTensor_Split(t *testing.T) { diff --git a/ts/patch.go b/ts/patch.go index 54e0f2b..42e8e31 100644 --- a/ts/patch.go +++ b/ts/patch.go @@ -7,7 +7,7 @@ import ( "log" "unsafe" - lib "github.com/sugarme/gotch/libtch" + lib "git.andr3h3nriqu3s.com/andr3/gotch/libtch" ) // NOTE. This is a temporarily patched to make it run. diff --git a/ts/print.go b/ts/print.go index c2e2b99..e98eac6 100644 --- a/ts/print.go +++ b/ts/print.go @@ -7,7 +7,7 @@ import ( "reflect" "strconv" - "github.com/sugarme/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch" ) func (ts *Tensor) ValueGo() interface{} { diff --git a/ts/print_test.go b/ts/print_test.go index 681d24d..eea2b6f 100644 --- a/ts/print_test.go +++ b/ts/print_test.go @@ -4,8 +4,8 @@ import ( "fmt" "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func TestTensor_Format(t *testing.T) { diff --git a/ts/scalar.go b/ts/scalar.go index d147d69..364a9a9 100644 --- a/ts/scalar.go +++ b/ts/scalar.go @@ -6,8 +6,8 @@ import ( "runtime" "sync/atomic" - "github.com/sugarme/gotch" - lib "github.com/sugarme/gotch/libtch" + "git.andr3h3nriqu3s.com/andr3/gotch" + lib "git.andr3h3nriqu3s.com/andr3/gotch/libtch" ) type Scalar struct { diff --git a/ts/tensor-generated.go b/ts/tensor-generated.go index 0077169..2e67d94 100644 --- a/ts/tensor-generated.go +++ b/ts/tensor-generated.go @@ -9,8 +9,8 @@ import( "unsafe" "fmt" - "github.com/sugarme/gotch" - lib "github.com/sugarme/gotch/libtch" + "git.andr3h3nriqu3s.com/andr3/gotch" + lib "git.andr3h3nriqu3s.com/andr3/gotch/libtch" ) diff --git a/ts/tensor.go b/ts/tensor.go index 61af2a0..2fba6ad 100644 --- a/ts/tensor.go +++ b/ts/tensor.go @@ -17,8 +17,8 @@ import ( "time" "unsafe" - gotch "github.com/sugarme/gotch" - lib "github.com/sugarme/gotch/libtch" + gotch "git.andr3h3nriqu3s.com/andr3/gotch" + lib "git.andr3h3nriqu3s.com/andr3/gotch/libtch" ) var ( diff --git a/ts/tensor_mem_test.go b/ts/tensor_mem_test.go index fca0c1c..24593bc 100644 --- a/ts/tensor_mem_test.go +++ b/ts/tensor_mem_test.go @@ -8,7 +8,7 @@ import ( "testing" "time" - "github.com/sugarme/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch" ) var n int = 10 diff --git a/ts/tensor_test.go b/ts/tensor_test.go index f82440b..263feef 100644 --- a/ts/tensor_test.go +++ b/ts/tensor_test.go @@ -4,8 +4,8 @@ import ( "reflect" "testing" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func TestTensorInit(t *testing.T) { diff --git a/ts/util.go b/ts/util.go index ec6efb2..2f41b12 100644 --- a/ts/util.go +++ b/ts/util.go @@ -12,7 +12,7 @@ import ( "reflect" "unsafe" - gotch "github.com/sugarme/gotch" + gotch "git.andr3h3nriqu3s.com/andr3/gotch" ) // nativeEndian is a ByteOrder for local platform. diff --git a/vision/alexnet.go b/vision/alexnet.go index 1b69406..b60fb58 100644 --- a/vision/alexnet.go +++ b/vision/alexnet.go @@ -1,8 +1,8 @@ package vision import ( - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // AlexNet implementation diff --git a/vision/aug/affine.go b/vision/aug/affine.go index d65b185..5be0d2b 100644 --- a/vision/aug/affine.go +++ b/vision/aug/affine.go @@ -1,8 +1,8 @@ package aug import ( - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // RandomAffine is transformation of the image keeping center invariant. diff --git a/vision/aug/blur.go b/vision/aug/blur.go index 0b31741..24a72d0 100644 --- a/vision/aug/blur.go +++ b/vision/aug/blur.go @@ -4,8 +4,8 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) type GaussianBlur struct { diff --git a/vision/aug/color.go b/vision/aug/color.go index c785d14..b7d6130 100644 --- a/vision/aug/color.go +++ b/vision/aug/color.go @@ -4,7 +4,7 @@ import ( "fmt" "log" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Ref. https://github.com/pytorch/vision/blob/f1d734213af65dc06e777877d315973ba8386080/torchvision/transforms/functional_tensor.py diff --git a/vision/aug/contrast.go b/vision/aug/contrast.go index cf84706..c99aad4 100644 --- a/vision/aug/contrast.go +++ b/vision/aug/contrast.go @@ -1,7 +1,7 @@ package aug import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // RandomAutocontrast autocontrasts the pixels of the given image randomly with a given probability. diff --git a/vision/aug/crop.go b/vision/aug/crop.go index 6532ae6..1025d38 100644 --- a/vision/aug/crop.go +++ b/vision/aug/crop.go @@ -6,8 +6,8 @@ import ( // "math" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) type RandomCrop struct { diff --git a/vision/aug/cutout.go b/vision/aug/cutout.go index 8959a50..e3001a7 100644 --- a/vision/aug/cutout.go +++ b/vision/aug/cutout.go @@ -5,8 +5,8 @@ import ( "log" "math" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Randomly selects a rectangle region in an torch Tensor image and erases its pixels. diff --git a/vision/aug/equalize.go b/vision/aug/equalize.go index 2f61d0d..950260d 100644 --- a/vision/aug/equalize.go +++ b/vision/aug/equalize.go @@ -1,7 +1,7 @@ package aug import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // RandomEqualize equalizes the histogram of the given image randomly with a given probability. diff --git a/vision/aug/flip.go b/vision/aug/flip.go index 091a0e6..33b3450 100644 --- a/vision/aug/flip.go +++ b/vision/aug/flip.go @@ -1,8 +1,8 @@ package aug import ( - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // RandomHorizontalFlip horizontally flips the given image randomly with a given probability. diff --git a/vision/aug/function.go b/vision/aug/function.go index 344c846..1cb9775 100644 --- a/vision/aug/function.go +++ b/vision/aug/function.go @@ -7,8 +7,8 @@ import ( "math/rand" "time" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func gaussianKernel1D(ks int64, sigma float64, dtype gotch.DType, device gotch.Device) *ts.Tensor { diff --git a/vision/aug/grayscale.go b/vision/aug/grayscale.go index 17f7266..d1d8f0e 100644 --- a/vision/aug/grayscale.go +++ b/vision/aug/grayscale.go @@ -3,7 +3,7 @@ package aug import ( "log" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // GrayScale converts image to grayscale. diff --git a/vision/aug/invert.go b/vision/aug/invert.go index 7248d30..1880603 100644 --- a/vision/aug/invert.go +++ b/vision/aug/invert.go @@ -1,7 +1,7 @@ package aug import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) type RandomInvert struct { diff --git a/vision/aug/normalize.go b/vision/aug/normalize.go index d413e38..51ee0fd 100644 --- a/vision/aug/normalize.go +++ b/vision/aug/normalize.go @@ -1,7 +1,7 @@ package aug import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Normalize normalizes a tensor image with mean and standard deviation. diff --git a/vision/aug/perspective.go b/vision/aug/perspective.go index 8b15c42..221474f 100644 --- a/vision/aug/perspective.go +++ b/vision/aug/perspective.go @@ -3,8 +3,8 @@ package aug import ( // "fmt" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // RandomPerspective performs a random perspective transformation of the given image with a given probability. diff --git a/vision/aug/posterize.go b/vision/aug/posterize.go index c07f39c..6f7325f 100644 --- a/vision/aug/posterize.go +++ b/vision/aug/posterize.go @@ -1,7 +1,7 @@ package aug import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // RandomPosterize posterizes the image randomly with a given probability by reducing the diff --git a/vision/aug/resize.go b/vision/aug/resize.go index a80e881..2ae6340 100644 --- a/vision/aug/resize.go +++ b/vision/aug/resize.go @@ -4,9 +4,9 @@ import ( "fmt" "log" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" - "github.com/sugarme/gotch/vision" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/vision" ) type ResizeModule struct { diff --git a/vision/aug/rotate.go b/vision/aug/rotate.go index b86cf8b..da5cdd8 100644 --- a/vision/aug/rotate.go +++ b/vision/aug/rotate.go @@ -7,8 +7,8 @@ import ( "math/rand" "time" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // RandomRotate randomly rotates a tensor image within a specifed angle range (degree). diff --git a/vision/aug/sharpness.go b/vision/aug/sharpness.go index 4725da4..f63ccd6 100644 --- a/vision/aug/sharpness.go +++ b/vision/aug/sharpness.go @@ -1,7 +1,7 @@ package aug import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Adjust the sharpness of the image randomly with a given probability. If the image is torch Tensor, diff --git a/vision/aug/solarize.go b/vision/aug/solarize.go index 0aa1ade..ab74e8e 100644 --- a/vision/aug/solarize.go +++ b/vision/aug/solarize.go @@ -1,7 +1,7 @@ package aug import ( - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // RandomSolarize solarizes the image randomly with a given probability by inverting all pixel diff --git a/vision/aug/transform.go b/vision/aug/transform.go index aaa0fc0..80aac7f 100644 --- a/vision/aug/transform.go +++ b/vision/aug/transform.go @@ -4,8 +4,8 @@ import ( "math/rand" "time" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Transformer is an interface that can transform an image tensor. diff --git a/vision/cifar.go b/vision/cifar.go index be9b1e1..4a9de5d 100644 --- a/vision/cifar.go +++ b/vision/cifar.go @@ -12,8 +12,8 @@ import ( "os" "path/filepath" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) const ( diff --git a/vision/dataset.go b/vision/dataset.go index 22fe9a6..8f57e11 100644 --- a/vision/dataset.go +++ b/vision/dataset.go @@ -8,7 +8,7 @@ import ( "math/rand" "time" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) type Dataset struct { diff --git a/vision/densenet.go b/vision/densenet.go index ae837ad..df048fb 100644 --- a/vision/densenet.go +++ b/vision/densenet.go @@ -8,8 +8,8 @@ package vision import ( "fmt" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func dnConv2d(p *nn.Path, cIn, cOut, ksize, padding, stride int64) *nn.Conv2D { diff --git a/vision/efficientnet.go b/vision/efficientnet.go index 695be34..5f7837a 100644 --- a/vision/efficientnet.go +++ b/vision/efficientnet.go @@ -4,8 +4,8 @@ import ( "fmt" "math" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) const ( diff --git a/vision/image.go b/vision/image.go index f4df9ad..a64858f 100644 --- a/vision/image.go +++ b/vision/image.go @@ -8,8 +8,8 @@ import ( "log" "math" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // (height, width, channel) -> (channel, height, width) diff --git a/vision/imagenet.go b/vision/imagenet.go index 64f3f38..5ee4840 100644 --- a/vision/imagenet.go +++ b/vision/imagenet.go @@ -10,8 +10,8 @@ import ( "reflect" "sync" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Helper functions for ImageNet like datasets. diff --git a/vision/inception.go b/vision/inception.go index 7b9ab52..b51533a 100644 --- a/vision/inception.go +++ b/vision/inception.go @@ -3,8 +3,8 @@ package vision // InceptionV3 import ( - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func convBn(p *nn.Path, cIn, cOut, ksize, pad, stride int64) ts.ModuleT { diff --git a/vision/mnist.go b/vision/mnist.go index 6d3ed87..b2b9aa5 100644 --- a/vision/mnist.go +++ b/vision/mnist.go @@ -11,8 +11,8 @@ import ( "os" "path/filepath" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // readInt32 read 4 bytes and convert to MSB first (big endian) interger. diff --git a/vision/mobilenet.go b/vision/mobilenet.go index 281c558..2799e73 100644 --- a/vision/mobilenet.go +++ b/vision/mobilenet.go @@ -6,9 +6,9 @@ package vision import ( "fmt" - "github.com/sugarme/gotch" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // Conv2D + BatchNorm2D + ReLU6 diff --git a/vision/resnet.go b/vision/resnet.go index 34a9452..a59dd11 100644 --- a/vision/resnet.go +++ b/vision/resnet.go @@ -3,8 +3,8 @@ package vision import ( "fmt" - nn "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + nn "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // ResNet implementation. diff --git a/vision/squeezenet.go b/vision/squeezenet.go index df129eb..0f26fe5 100644 --- a/vision/squeezenet.go +++ b/vision/squeezenet.go @@ -3,8 +3,8 @@ package vision // SqueezeNet implementation. import ( - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) func snMaxPool2D(xs *ts.Tensor) *ts.Tensor { diff --git a/vision/vgg.go b/vision/vgg.go index 4550785..db80c99 100644 --- a/vision/vgg.go +++ b/vision/vgg.go @@ -5,8 +5,8 @@ package vision import ( "fmt" - "github.com/sugarme/gotch/nn" - "github.com/sugarme/gotch/ts" + "git.andr3h3nriqu3s.com/andr3/gotch/nn" + "git.andr3h3nriqu3s.com/andr3/gotch/ts" ) // NOTE: each list element contains multiple convolutions with some specified number