torchvision 0.4.0 → 0.4.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/CHANGELOG.md +4 -0
- data/LICENSE.txt +1 -1
- data/lib/torchvision/datasets/kmnist.rb +1 -1
- data/lib/torchvision/datasets/mnist.rb +3 -3
- data/lib/torchvision/models/resnet.rb +10 -3
- data/lib/torchvision/version.rb +1 -1
- metadata +3 -7
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA256:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: ed3dfc1059059e8ffdd6d0e2079d77f867ab484bfd4801e39eb0641818bcc1ed
|
|
4
|
+
data.tar.gz: 16a40098b804d6dc35738b78b0693f6ebffaca7b0d85016dcca95c4f20c45627
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
6
|
+
metadata.gz: a1ed0b3e10acee4469a4b772a6cbcba164e697d2a51e2225324f558183310d455d5d28b6a722c5c32ce2a26940d22d94ecbdddaaa25c08994a6f69a30f731b7c
|
|
7
|
+
data.tar.gz: b2f2fd94a3bd183edee1467dad2e44a53535d6e0cbd922b553d5b2e9cb9a6b3a1bdc935ec5dc2bf6fd97c4f131b35eb09fc6a8339c5eba417fe696f64fee1e6b
|
data/CHANGELOG.md
CHANGED
data/LICENSE.txt
CHANGED
|
@@ -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, Net::HTTPClientException => e
|
|
62
|
+
rescue Errno::ECONNREFUSED, Net::HTTPFatalError, Net::HTTPClientException => e
|
|
63
63
|
puts "Failed to download (trying next): #{e.message}"
|
|
64
64
|
end
|
|
65
65
|
end
|
|
@@ -87,8 +87,8 @@ module TorchVision
|
|
|
87
87
|
|
|
88
88
|
def mirrors
|
|
89
89
|
[
|
|
90
|
-
"https://
|
|
91
|
-
"https://
|
|
90
|
+
"https://ossci-datasets.s3.amazonaws.com/mnist/",
|
|
91
|
+
"https://yann.lecun.com/exdb/mnist/"
|
|
92
92
|
]
|
|
93
93
|
end
|
|
94
94
|
|
|
@@ -13,9 +13,16 @@ module TorchVision
|
|
|
13
13
|
"wide_resnet101_2" => "https://download.pytorch.org/models/wide_resnet101_2-32ee1156.pth"
|
|
14
14
|
}
|
|
15
15
|
|
|
16
|
-
def initialize(
|
|
17
|
-
|
|
18
|
-
|
|
16
|
+
def initialize(
|
|
17
|
+
block,
|
|
18
|
+
layers,
|
|
19
|
+
num_classes = 1000,
|
|
20
|
+
zero_init_residual: false,
|
|
21
|
+
groups: 1,
|
|
22
|
+
width_per_group: 64,
|
|
23
|
+
replace_stride_with_dilation: nil,
|
|
24
|
+
norm_layer: nil
|
|
25
|
+
)
|
|
19
26
|
super()
|
|
20
27
|
norm_layer ||= Torch::NN::BatchNorm2d
|
|
21
28
|
@norm_layer = norm_layer
|
data/lib/torchvision/version.rb
CHANGED
metadata
CHANGED
|
@@ -1,14 +1,13 @@
|
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
|
2
2
|
name: torchvision
|
|
3
3
|
version: !ruby/object:Gem::Version
|
|
4
|
-
version: 0.4.
|
|
4
|
+
version: 0.4.1
|
|
5
5
|
platform: ruby
|
|
6
6
|
authors:
|
|
7
7
|
- Andrew Kane
|
|
8
|
-
autorequire:
|
|
9
8
|
bindir: bin
|
|
10
9
|
cert_chain: []
|
|
11
|
-
date:
|
|
10
|
+
date: 1980-01-02 00:00:00.000000000 Z
|
|
12
11
|
dependencies:
|
|
13
12
|
- !ruby/object:Gem::Dependency
|
|
14
13
|
name: numo-narray
|
|
@@ -52,7 +51,6 @@ dependencies:
|
|
|
52
51
|
- - ">="
|
|
53
52
|
- !ruby/object:Gem::Version
|
|
54
53
|
version: '0.13'
|
|
55
|
-
description:
|
|
56
54
|
email: andrew@ankane.org
|
|
57
55
|
executables: []
|
|
58
56
|
extensions: []
|
|
@@ -107,7 +105,6 @@ homepage: https://github.com/ankane/torchvision-ruby
|
|
|
107
105
|
licenses:
|
|
108
106
|
- BSD-3-Clause
|
|
109
107
|
metadata: {}
|
|
110
|
-
post_install_message:
|
|
111
108
|
rdoc_options: []
|
|
112
109
|
require_paths:
|
|
113
110
|
- lib
|
|
@@ -122,8 +119,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
|
|
|
122
119
|
- !ruby/object:Gem::Version
|
|
123
120
|
version: '0'
|
|
124
121
|
requirements: []
|
|
125
|
-
rubygems_version: 3.
|
|
126
|
-
signing_key:
|
|
122
|
+
rubygems_version: 3.6.9
|
|
127
123
|
specification_version: 4
|
|
128
124
|
summary: Computer vision datasets, transforms, and models for Ruby
|
|
129
125
|
test_files: []
|