File tree 12 files changed +16
-17
lines changed
tutorials/transfer_learning
12 files changed +16
-17
lines changed Original file line number Diff line number Diff line change @@ -4,5 +4,5 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
4
4
Statistics = " 10745b16-79ce-11e8-11f9-7d13ad32a3b2"
5
5
6
6
[compat ]
7
- Flux = " 0.13.9"
8
- julia = " 1.6"
7
+ Flux = " 0.13.9, 0.14 "
8
+ julia = " 1.6"
Original file line number Diff line number Diff line change @@ -3,5 +3,5 @@ Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
3
3
Test = " 8dfed614-e22c-5e08-85e1-65c5234f0b40"
4
4
5
5
[compat ]
6
- Flux = " 0.13.9"
6
+ Flux = " 0.13.9, 0.14 "
7
7
julia = " 1.6"
Original file line number Diff line number Diff line change @@ -6,6 +6,6 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
6
6
7
7
[compat ]
8
8
DataFrames = " 1.4.3"
9
- Flux = " 0.13.9"
9
+ Flux = " 0.13.9, 0.14 "
10
10
MLDatasets = " 0.7.6"
11
11
julia = " 1.6"
Original file line number Diff line number Diff line change @@ -4,5 +4,5 @@ OneHotArrays = "0b1bfda6-eb8a-41d2-88d8-f5af5cad476f"
4
4
StatsBase = " 2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
5
5
6
6
[compat ]
7
- Flux = " 0.13.9"
7
+ Flux = " 0.13.9, 0.14 "
8
8
julia = " 1.6"
Original file line number Diff line number Diff line change @@ -9,7 +9,7 @@ Unicode = "4ec0a83e-493e-50e2-b9ac-8f72acf5a8f5"
9
9
10
10
[compat ]
11
11
Cascadia = " 1"
12
- Flux = " 0.13.9"
12
+ Flux = " 0.13.9, 0.14 "
13
13
Gumbo = " 0.8"
14
14
HTTP = " 1.7"
15
15
julia = " 1.6"
Original file line number Diff line number Diff line change @@ -7,5 +7,5 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
7
7
Statistics = " 10745b16-79ce-11e8-11f9-7d13ad32a3b2"
8
8
9
9
[compat ]
10
- Flux = " 0.13"
10
+ Flux = " 0.13, 0.14 "
11
11
julia = " 1.6"
Original file line number Diff line number Diff line change 2
2
using Random: shuffle!
3
3
import Base: length, getindex
4
4
using Images
5
- using Flux
5
+ using Flux, CUDA
6
6
using Flux: update!
7
7
using DataAugmentation
8
8
using Metalhead
9
9
10
- device = Flux . CUDA. functional () ? gpu : cpu
10
+ device = CUDA. functional () ? gpu : cpu
11
11
# device = cpu
12
12
13
13
# # Custom DataLoader
Original file line number Diff line number Diff line change @@ -7,7 +7,6 @@ MLDatasets = "eb30cadb-4394-5ae3-aed4-317e484a6458"
7
7
Statistics = " 10745b16-79ce-11e8-11f9-7d13ad32a3b2"
8
8
9
9
[compat ]
10
- CUDA = " 2.4.0, 3"
11
- Flux = " 0.13.9"
10
+ Flux = " 0.13.9, 0.14"
12
11
MLDatasets = " 0.7"
13
12
julia = " 1.6"
Original file line number Diff line number Diff line change @@ -7,6 +7,6 @@ Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
7
7
Statistics = " 10745b16-79ce-11e8-11f9-7d13ad32a3b2"
8
8
9
9
[compat ]
10
- Flux = " 0.13.9"
10
+ Flux = " 0.13.9, 0.14 "
11
11
MLDatasets = " 0.7"
12
12
julia = " 1.6"
Original file line number Diff line number Diff line change @@ -13,6 +13,6 @@ ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444"
13
13
14
14
[compat ]
15
15
DrWatson = " 2.7.6"
16
- Flux = " 0.13.9"
16
+ Flux = " 0.13.9, 0.14 "
17
17
MLDatasets = " 0.7.6"
18
18
julia = " 1.7.0"
Original file line number Diff line number Diff line change @@ -12,7 +12,7 @@ ProgressMeter = "92933f4c-e287-5a05-a399-4b506db050ca"
12
12
TensorBoardLogger = " 899adc3e-224a-11e9-021f-63837185c80f"
13
13
14
14
[compat ]
15
- CUDA = " 2.4, 3.12"
16
- Flux = " 0.13.9"
15
+ CUDA = " 2.4, 3.12, 4 "
16
+ Flux = " 0.13.9, 0.14 "
17
17
MLDatasets = " 0.7"
18
18
julia = " 1.6"
Original file line number Diff line number Diff line change @@ -7,8 +7,8 @@ MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54"
7
7
Statistics = " 10745b16-79ce-11e8-11f9-7d13ad32a3b2"
8
8
9
9
[compat ]
10
- CUDA = " 3"
11
- Flux = " 0.13.9"
10
+ CUDA = " 3, 4 "
11
+ Flux = " 0.13.9, 0.14 "
12
12
MLDatasets = " 0.7"
13
13
MLUtils = " 0.3"
14
14
julia = " 1.6"
You can’t perform that action at this time.
0 commit comments