torchvision 0.3.0 → 0.4.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/CHANGELOG.md +4 -0
- data/LICENSE.txt +1 -1
- data/README.md +1 -1
- data/lib/torchvision/datasets/cifar10.rb +1 -1
- data/lib/torchvision/datasets/mnist.rb +2 -2
- data/lib/torchvision/models/resnet.rb +1 -1
- data/lib/torchvision/models/vgg.rb +1 -1
- data/lib/torchvision/version.rb +1 -1
- metadata +6 -6
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 73e4a86f8dfcd27a7fe92c82f68865116b8d0906f07c1b1887a3a2990b5b9f1d
|
4
|
+
data.tar.gz: 20dfe4a76998626a75542048fee9f9529e14cb85cc7935be3b66fa521751ae21
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: f5e4383b65158f675cce3bc14dba875acd2182e59db0b005a9d288c58189b52bea2698ed60349e16f04e207e8a1be13a8278a6d1633d70db6bc2a6a2c3c38165
|
7
|
+
data.tar.gz: 70ddf739f2f85eeb8ce78bb098930aae169497544ab996b152800ca7b06333bc9e2ed541834f472145dbe82582c77f11c0af2404c9e5cea329b80bd25348d6ab
|
data/CHANGELOG.md
CHANGED
data/LICENSE.txt
CHANGED
data/README.md
CHANGED
@@ -2,7 +2,7 @@
|
|
2
2
|
|
3
3
|
:fire: Computer vision datasets, transforms, and models for Ruby
|
4
4
|
|
5
|
-
[![Build Status](https://github.com/ankane/torchvision-ruby/workflows/build/badge.svg
|
5
|
+
[![Build Status](https://github.com/ankane/torchvision-ruby/actions/workflows/build.yml/badge.svg)](https://github.com/ankane/torchvision-ruby/actions)
|
6
6
|
|
7
7
|
## Installation
|
8
8
|
|
@@ -99,7 +99,7 @@ module TorchVision
|
|
99
99
|
{filename: "data_batch_2.bin", sha256: "a591ca11fa1708a91ee40f54b3da4784ccd871ecf2137de63f51ada8b3fa57ed"},
|
100
100
|
{filename: "data_batch_3.bin", sha256: "bbe8596564c0f86427f876058170b84dac6670ddf06d79402899d93ceea26f67"},
|
101
101
|
{filename: "data_batch_4.bin", sha256: "014e562d6e23c72197cc727519169a60359f5eccd8945ad5a09d710285ff4e48"},
|
102
|
-
{filename: "data_batch_5.bin", sha256: "755304fc0b379caeae8c14f0dac912fbc7d6cd469eb67a1029a08a39453a9add"}
|
102
|
+
{filename: "data_batch_5.bin", sha256: "755304fc0b379caeae8c14f0dac912fbc7d6cd469eb67a1029a08a39453a9add"}
|
103
103
|
]
|
104
104
|
end
|
105
105
|
|
@@ -59,7 +59,7 @@ module TorchVision
|
|
59
59
|
download_file(url, download_root: raw_folder, filename: resource[:filename], sha256: resource[:sha256])
|
60
60
|
success = true
|
61
61
|
break
|
62
|
-
rescue Net::HTTPFatalError => e
|
62
|
+
rescue Net::HTTPFatalError, Net::HTTPClientException => e
|
63
63
|
puts "Failed to download (trying next): #{e.message}"
|
64
64
|
end
|
65
65
|
end
|
@@ -87,7 +87,7 @@ module TorchVision
|
|
87
87
|
|
88
88
|
def mirrors
|
89
89
|
[
|
90
|
-
"
|
90
|
+
"https://yann.lecun.com/exdb/mnist/",
|
91
91
|
"https://ossci-datasets.s3.amazonaws.com/mnist/"
|
92
92
|
]
|
93
93
|
end
|
@@ -13,7 +13,7 @@ module TorchVision
|
|
13
13
|
"wide_resnet101_2" => "https://download.pytorch.org/models/wide_resnet101_2-32ee1156.pth"
|
14
14
|
}
|
15
15
|
|
16
|
-
def initialize(block, layers, num_classes=1000, zero_init_residual: false,
|
16
|
+
def initialize(block, layers, num_classes = 1000, zero_init_residual: false,
|
17
17
|
groups: 1, width_per_group: 64, replace_stride_with_dilation: nil, norm_layer: nil)
|
18
18
|
|
19
19
|
super()
|
@@ -56,7 +56,7 @@ module TorchVision
|
|
56
56
|
"A" => [64, "M", 128, "M", 256, 256, "M", 512, 512, "M", 512, 512, "M"],
|
57
57
|
"B" => [64, 64, "M", 128, 128, "M", 256, 256, "M", 512, 512, "M", 512, 512, "M"],
|
58
58
|
"D" => [64, 64, "M", 128, 128, "M", 256, 256, 256, "M", 512, 512, 512, "M", 512, 512, 512, "M"],
|
59
|
-
"E" => [64, 64, "M", 128, 128, "M", 256, 256, 256, 256, "M", 512, 512, 512, 512, "M", 512, 512, 512, 512, "M"]
|
59
|
+
"E" => [64, 64, "M", 128, 128, "M", 256, 256, 256, 256, "M", 512, 512, 512, 512, "M", 512, 512, 512, 512, "M"]
|
60
60
|
}
|
61
61
|
|
62
62
|
def self.make_model(arch, cfg, batch_norm, pretrained: false, **kwargs)
|
data/lib/torchvision/version.rb
CHANGED
metadata
CHANGED
@@ -1,14 +1,14 @@
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
2
2
|
name: torchvision
|
3
3
|
version: !ruby/object:Gem::Version
|
4
|
-
version: 0.
|
4
|
+
version: 0.4.0
|
5
5
|
platform: ruby
|
6
6
|
authors:
|
7
7
|
- Andrew Kane
|
8
8
|
autorequire:
|
9
9
|
bindir: bin
|
10
10
|
cert_chain: []
|
11
|
-
date:
|
11
|
+
date: 2024-08-03 00:00:00.000000000 Z
|
12
12
|
dependencies:
|
13
13
|
- !ruby/object:Gem::Dependency
|
14
14
|
name: numo-narray
|
@@ -44,14 +44,14 @@ dependencies:
|
|
44
44
|
requirements:
|
45
45
|
- - ">="
|
46
46
|
- !ruby/object:Gem::Version
|
47
|
-
version: 0.
|
47
|
+
version: '0.13'
|
48
48
|
type: :runtime
|
49
49
|
prerelease: false
|
50
50
|
version_requirements: !ruby/object:Gem::Requirement
|
51
51
|
requirements:
|
52
52
|
- - ">="
|
53
53
|
- !ruby/object:Gem::Version
|
54
|
-
version: 0.
|
54
|
+
version: '0.13'
|
55
55
|
description:
|
56
56
|
email: andrew@ankane.org
|
57
57
|
executables: []
|
@@ -115,14 +115,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
|
|
115
115
|
requirements:
|
116
116
|
- - ">="
|
117
117
|
- !ruby/object:Gem::Version
|
118
|
-
version: '
|
118
|
+
version: '3.1'
|
119
119
|
required_rubygems_version: !ruby/object:Gem::Requirement
|
120
120
|
requirements:
|
121
121
|
- - ">="
|
122
122
|
- !ruby/object:Gem::Version
|
123
123
|
version: '0'
|
124
124
|
requirements: []
|
125
|
-
rubygems_version: 3.
|
125
|
+
rubygems_version: 3.5.11
|
126
126
|
signing_key:
|
127
127
|
specification_version: 4
|
128
128
|
summary: Computer vision datasets, transforms, and models for Ruby
|