torchvision 0.3.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.
Files changed (11) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +8 -0
  3. data/LICENSE.txt +1 -1
  4. data/README.md +1 -1
  5. data/lib/torchvision/datasets/cifar10.rb +1 -1
  6. data/lib/torchvision/datasets/kmnist.rb +1 -1
  7. data/lib/torchvision/datasets/mnist.rb +3 -3
  8. data/lib/torchvision/models/resnet.rb +10 -3
  9. data/lib/torchvision/models/vgg.rb +1 -1
  10. data/lib/torchvision/version.rb +1 -1
  11. metadata +6 -10
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: 4d83605afd88b375cddc9a57fed3e1c6f7654bd9657623a54c99d4b1d8f0ecb5
4
- data.tar.gz: '09bf57d203ac64aceab4d4cbe2f72de93087d5f588a17fa37ac80b93c34a371b'
3
+ metadata.gz: ed3dfc1059059e8ffdd6d0e2079d77f867ab484bfd4801e39eb0641818bcc1ed
4
+ data.tar.gz: 16a40098b804d6dc35738b78b0693f6ebffaca7b0d85016dcca95c4f20c45627
5
5
  SHA512:
6
- metadata.gz: a890bdeaca6a26af1d245b1b60d4e228b066303393fa25b6c08d83f2253e99c662fba5050205156d6dc80382531cb2236bdbaceb591305c227f66e61a9e5fa7b
7
- data.tar.gz: 6d31e7adfc560c1571f1a986dc9f60cab62e00659f582321d52cff731898864ec3e47dbe476a7314e6df9633941475502e412a08a87f270c7e21739b5280f776
6
+ metadata.gz: a1ed0b3e10acee4469a4b772a6cbcba164e697d2a51e2225324f558183310d455d5d28b6a722c5c32ce2a26940d22d94ecbdddaaa25c08994a6f69a30f731b7c
7
+ data.tar.gz: b2f2fd94a3bd183edee1467dad2e44a53535d6e0cbd922b553d5b2e9cb9a6b3a1bdc935ec5dc2bf6fd97c4f131b35eb09fc6a8339c5eba417fe696f64fee1e6b
data/CHANGELOG.md CHANGED
@@ -1,3 +1,11 @@
1
+ ## 0.4.1 (2025-08-07)
2
+
3
+ - Fixed download for MNIST and KMNIST datasets
4
+
5
+ ## 0.4.0 (2024-08-02)
6
+
7
+ - Dropped support for Ruby < 3.1
8
+
1
9
  ## 0.3.0 (2023-01-29)
2
10
 
3
11
  - Added mirror for MNIST dataset
data/LICENSE.txt CHANGED
@@ -1,7 +1,7 @@
1
1
  BSD 3-Clause License
2
2
 
3
3
  Copyright (c) Soumith Chintala 2016,
4
- Copyright (c) Andrew Kane 2020-2023,
4
+ Copyright (c) Andrew Kane 2020-2025,
5
5
  All rights reserved.
6
6
 
7
7
  Redistribution and use in source and binary forms, with or without
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?branch=master)](https://github.com/ankane/torchvision-ruby/actions)
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
 
data/lib/torchvision/datasets/cifar10.rb CHANGED
@@ -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
 
data/lib/torchvision/datasets/kmnist.rb CHANGED
@@ -7,7 +7,7 @@ module TorchVision
7
7
 
8
8
  def mirrors
9
9
  [
10
- "http://codh.rois.ac.jp/kmnist/dataset/kmnist/"
10
+ "https://codh.rois.ac.jp/kmnist/dataset/kmnist/"
11
11
  ]
12
12
  end
13
13
 
data/lib/torchvision/datasets/mnist.rb 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 => 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
- "http://yann.lecun.com/exdb/mnist/",
91
- "https://ossci-datasets.s3.amazonaws.com/mnist/"
90
+ "https://ossci-datasets.s3.amazonaws.com/mnist/",
91
+ "https://yann.lecun.com/exdb/mnist/"
92
92
  ]
93
93
  end
94
94
 
data/lib/torchvision/models/resnet.rb CHANGED
@@ -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(block, layers, num_classes=1000, zero_init_residual: false,
17
- groups: 1, width_per_group: 64, replace_stride_with_dilation: nil, norm_layer: nil)
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/models/vgg.rb CHANGED
@@ -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
@@ -1,3 +1,3 @@
1
1
  module TorchVision
2
- VERSION = "0.3.0"
2
+ VERSION = "0.4.1"
3
3
  end
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.3.0
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: 2023-01-30 00:00:00.000000000 Z
10
+ date: 1980-01-02 00:00:00.000000000 Z
12
11
  dependencies:
13
12
  - !ruby/object:Gem::Dependency
14
13
  name: numo-narray
@@ -44,15 +43,14 @@ dependencies:
44
43
  requirements:
45
44
  - - ">="
46
45
  - !ruby/object:Gem::Version
47
- version: 0.11.1
46
+ version: '0.13'
48
47
  type: :runtime
49
48
  prerelease: false
50
49
  version_requirements: !ruby/object:Gem::Requirement
51
50
  requirements:
52
51
  - - ">="
53
52
  - !ruby/object:Gem::Version
54
- version: 0.11.1
55
- description:
53
+ version: '0.13'
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
@@ -115,15 +112,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
115
112
  requirements:
116
113
  - - ">="
117
114
  - !ruby/object:Gem::Version
118
- version: '2.7'
115
+ version: '3.1'
119
116
  required_rubygems_version: !ruby/object:Gem::Requirement
120
117
  requirements:
121
118
  - - ">="
122
119
  - !ruby/object:Gem::Version
123
120
  version: '0'
124
121
  requirements: []
125
- rubygems_version: 3.4.1
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: []