Last active
February 8, 2023 06:39
-
-
Save RomeoV/ca397a6b883c1cf567f2503d135084d8 to your computer and use it in GitHub Desktop.
Bugreport PyCallChainRules.jl Issue 29
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# This file is machine-generated - editing it directly is not advised | |
julia_version = "1.9.0-beta3" | |
manifest_format = "2.0" | |
project_hash = "0dcb1be53cdfb37b6d1a06796f509eec646f7a67" | |
[[deps.AbstractFFTs]] | |
deps = ["ChainRulesCore", "LinearAlgebra"] | |
git-tree-sha1 = "69f7020bd72f069c219b5e8c236c1fa90d2cb409" | |
uuid = "621f4979-c628-5d54-868e-fcf4e3e8185c" | |
version = "1.2.1" | |
[[deps.AbstractTrees]] | |
git-tree-sha1 = "faa260e4cb5aba097a73fab382dd4b5819d8ec8c" | |
uuid = "1520ce14-60c1-5f80-bbc7-55ef81b5835c" | |
version = "0.4.4" | |
[[deps.Accessors]] | |
deps = ["Compat", "CompositionsBase", "ConstructionBase", "Dates", "InverseFunctions", "LinearAlgebra", "MacroTools", "Requires", "Test"] | |
git-tree-sha1 = "b9661b900b50ba475145b311a9a0ef9d2a9c85ea" | |
uuid = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697" | |
version = "0.1.26" | |
weakdeps = ["StaticArrays"] | |
[deps.Accessors.extensions] | |
StaticArraysExt = "StaticArrays" | |
[[deps.Adapt]] | |
deps = ["LinearAlgebra"] | |
git-tree-sha1 = "0310e08cb19f5da31d08341c6120c047598f5b9c" | |
uuid = "79e6a3ab-5dfb-504d-930d-738a2a938a0e" | |
version = "3.5.0" | |
[[deps.ArgCheck]] | |
git-tree-sha1 = "a3a402a35a2f7e0b87828ccabbd5ebfbebe356b4" | |
uuid = "dce04be8-c92d-5529-be00-80e4d2c0e197" | |
version = "2.3.0" | |
[[deps.ArgTools]] | |
uuid = "0dad84c5-d112-42e6-8d28-ef12dabb789f" | |
version = "1.1.1" | |
[[deps.ArnoldiMethod]] | |
deps = ["LinearAlgebra", "Random", "StaticArrays"] | |
git-tree-sha1 = "62e51b39331de8911e4a7ff6f5aaf38a5f4cc0ae" | |
uuid = "ec485272-7323-5ecc-a04f-4719b315124d" | |
version = "0.2.0" | |
[[deps.ArrayLayouts]] | |
deps = ["FillArrays", "LinearAlgebra", "SparseArrays"] | |
git-tree-sha1 = "4aff5fa660eb95c2e0deb6bcdabe4d9a96bc4667" | |
uuid = "4c555306-a7a7-4459-81d9-ec55ddd5c99a" | |
version = "0.8.18" | |
[[deps.Artifacts]] | |
uuid = "56f22d72-fd6d-98f1-02f0-08ddc0907c33" | |
[[deps.AxisAlgorithms]] | |
deps = ["LinearAlgebra", "Random", "SparseArrays", "WoodburyMatrices"] | |
git-tree-sha1 = "66771c8d21c8ff5e3a93379480a2307ac36863f7" | |
uuid = "13072b0f-2c55-5437-9ae7-d433b7a33950" | |
version = "1.0.1" | |
[[deps.AxisArrays]] | |
deps = ["Dates", "IntervalSets", "IterTools", "RangeArrays"] | |
git-tree-sha1 = "1dd4d9f5beebac0c03446918741b1a03dc5e5788" | |
uuid = "39de3d68-74b9-583c-8d2d-e117c070f3a9" | |
version = "0.4.6" | |
[[deps.BFloat16s]] | |
deps = ["LinearAlgebra", "Printf", "Random", "Test"] | |
git-tree-sha1 = "a598ecb0d717092b5539dbbe890c98bac842b072" | |
uuid = "ab4f0b2a-ad5b-11e8-123f-65d77653426b" | |
version = "0.2.0" | |
[[deps.BSON]] | |
git-tree-sha1 = "86e9781ac28f4e80e9b98f7f96eae21891332ac2" | |
uuid = "fbb218c0-5317-5bc6-957e-2ee96dd4b1f0" | |
version = "0.3.6" | |
[[deps.BangBang]] | |
deps = ["Compat", "ConstructionBase", "Future", "InitialValues", "LinearAlgebra", "Requires", "Setfield", "Tables", "ZygoteRules"] | |
git-tree-sha1 = "7fe6d92c4f281cf4ca6f2fba0ce7b299742da7ca" | |
uuid = "198e06fe-97b7-11e9-32a5-e1d131e6ad66" | |
version = "0.3.37" | |
[[deps.Base64]] | |
uuid = "2a0f44e3-6c83-55bd-87e4-b1978d98bd5f" | |
[[deps.Baselet]] | |
git-tree-sha1 = "aebf55e6d7795e02ca500a689d326ac979aaf89e" | |
uuid = "9718e550-a3fa-408a-8086-8db961cd8217" | |
version = "0.1.1" | |
[[deps.BinaryProvider]] | |
deps = ["Libdl", "Logging", "SHA"] | |
git-tree-sha1 = "ecdec412a9abc8db54c0efc5548c64dfce072058" | |
uuid = "b99e7846-7c00-51b0-8f62-c81ae34c0232" | |
version = "0.5.10" | |
[[deps.BitFlags]] | |
git-tree-sha1 = "43b1a4a8f797c1cddadf60499a8a077d4af2cd2d" | |
uuid = "d1d4a3ce-64b1-5f1a-9ba4-7e7e69966f35" | |
version = "0.1.7" | |
[[deps.BufferedStreams]] | |
git-tree-sha1 = "bb065b14d7f941b8617bc323063dbe79f55d16ea" | |
uuid = "e1450e63-4bb3-523b-b2a4-4ffa8c0fd77d" | |
version = "1.1.0" | |
[[deps.Bzip2_jll]] | |
deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg"] | |
git-tree-sha1 = "19a35467a82e236ff51bc17a3a44b69ef35185a2" | |
uuid = "6e34b625-4abd-537c-b88f-471c36dfa7a0" | |
version = "1.0.8+0" | |
[[deps.CEnum]] | |
git-tree-sha1 = "eb4cb44a499229b3b8426dcfb5dd85333951ff90" | |
uuid = "fa961155-64e5-5f13-b03f-caf6b980ea82" | |
version = "0.4.2" | |
[[deps.CRC32c]] | |
uuid = "8bf52ea8-c179-5cab-976a-9e18b702a9bc" | |
[[deps.CSV]] | |
deps = ["CodecZlib", "Dates", "FilePathsBase", "InlineStrings", "Mmap", "Parsers", "PooledArrays", "SentinelArrays", "SnoopPrecompile", "Tables", "Unicode", "WeakRefStrings", "WorkerUtilities"] | |
git-tree-sha1 = "c700cce799b51c9045473de751e9319bdd1c6e94" | |
uuid = "336ed68f-0bac-5ca0-87d4-7b16caf5d00b" | |
version = "0.10.9" | |
[[deps.CUDA]] | |
deps = ["AbstractFFTs", "Adapt", "BFloat16s", "CEnum", "CompilerSupportLibraries_jll", "ExprTools", "GPUArrays", "GPUCompiler", "LLVM", "LazyArtifacts", "Libdl", "LinearAlgebra", "Logging", "Printf", "Random", "Random123", "RandomNumbers", "Reexport", "Requires", "SparseArrays", "SpecialFunctions", "TimerOutputs"] | |
git-tree-sha1 = "6717cb9a3425ebb7b31ca4f832823615d175f64a" | |
uuid = "052768ef-5323-5732-b1bb-66c8b64840ba" | |
version = "3.13.1" | |
[[deps.Calculus]] | |
deps = ["LinearAlgebra"] | |
git-tree-sha1 = "f641eb0a4f00c343bbc32346e1217b86f3ce9dad" | |
uuid = "49dc2e85-a5d0-5ad3-a950-438e2897f1b9" | |
version = "0.5.1" | |
[[deps.ChainRules]] | |
deps = ["ChainRulesCore", "Compat", "Distributed", "GPUArraysCore", "IrrationalConstants", "LinearAlgebra", "Random", "RealDot", "SparseArrays", "Statistics"] | |
git-tree-sha1 = "f9d6dd293ed05233d37a5644f880f5def9fdfae3" | |
uuid = "082447d4-558c-5d27-93f4-14fc19e9eca2" | |
version = "1.42.0" | |
[[deps.ChainRulesCore]] | |
deps = ["Compat", "LinearAlgebra", "SparseArrays"] | |
git-tree-sha1 = "c6d890a52d2c4d55d326439580c3b8d0875a77d9" | |
uuid = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4" | |
version = "1.15.7" | |
[[deps.Chemfiles]] | |
deps = ["BinaryProvider", "Chemfiles_jll", "DocStringExtensions", "Libdl"] | |
git-tree-sha1 = "3b4a49a0a4c9b2ff8c0c6ec035c16f32955531a8" | |
uuid = "46823bd8-5fb3-5f92-9aa0-96921f3dd015" | |
version = "0.10.3" | |
[[deps.Chemfiles_jll]] | |
deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg"] | |
git-tree-sha1 = "d4e54b053fc584e7a0f37e9d3a5c4500927b343a" | |
uuid = "78a364fa-1a3c-552a-b4bb-8fa0f9c1fcca" | |
version = "0.10.3+0" | |
[[deps.CodecZlib]] | |
deps = ["TranscodingStreams", "Zlib_jll"] | |
git-tree-sha1 = "9c209fb7536406834aa938fb149964b985de6c83" | |
uuid = "944b1d66-785c-5afd-91f1-9de20f533193" | |
version = "0.7.1" | |
[[deps.ColorBlendModes]] | |
deps = ["ColorTypes", "FixedPointNumbers"] | |
git-tree-sha1 = "9ec825436862d5ab02ad8f2cde72f2d860151fa6" | |
uuid = "60508b50-96e1-4007-9d6c-f475c410f16b" | |
version = "0.2.4" | |
[[deps.ColorSchemes]] | |
deps = ["ColorTypes", "ColorVectorSpace", "Colors", "FixedPointNumbers", "Random", "SnoopPrecompile"] | |
git-tree-sha1 = "aa3edc8f8dea6cbfa176ee12f7c2fc82f0608ed3" | |
uuid = "35d6a980-a343-548e-a6ea-1d62b119f2f4" | |
version = "3.20.0" | |
[[deps.ColorTypes]] | |
deps = ["FixedPointNumbers", "Random"] | |
git-tree-sha1 = "eb7f0f8307f71fac7c606984ea5fb2817275d6e4" | |
uuid = "3da002f7-5984-5a60-b8a6-cbb66c0b333f" | |
version = "0.11.4" | |
[[deps.ColorVectorSpace]] | |
deps = ["ColorTypes", "FixedPointNumbers", "LinearAlgebra", "SpecialFunctions", "Statistics", "TensorCore"] | |
git-tree-sha1 = "600cc5508d66b78aae350f7accdb58763ac18589" | |
uuid = "c3611d14-8923-5661-9e6a-0046d554d3a4" | |
version = "0.9.10" | |
[[deps.Colors]] | |
deps = ["ColorTypes", "FixedPointNumbers", "Reexport"] | |
git-tree-sha1 = "fc08e5930ee9a4e03f84bfb5211cb54e7769758a" | |
uuid = "5ae59095-9a9b-59fe-a467-6f913c188581" | |
version = "0.12.10" | |
[[deps.CommonSubexpressions]] | |
deps = ["MacroTools", "Test"] | |
git-tree-sha1 = "7b8a93dba8af7e3b42fecabf646260105ac373f7" | |
uuid = "bbf7d656-a473-5ed7-a52c-81e309532950" | |
version = "0.3.0" | |
[[deps.Compat]] | |
deps = ["Dates", "LinearAlgebra", "UUIDs"] | |
git-tree-sha1 = "61fdd77467a5c3ad071ef8277ac6bd6af7dd4c04" | |
uuid = "34da2185-b29b-5c13-b0c7-acf172513d20" | |
version = "4.6.0" | |
[[deps.CompilerSupportLibraries_jll]] | |
deps = ["Artifacts", "Libdl"] | |
uuid = "e66e0078-7015-5450-92f7-15fbd957f2ae" | |
version = "1.0.2+0" | |
[[deps.CompositionsBase]] | |
git-tree-sha1 = "455419f7e328a1a2493cabc6428d79e951349769" | |
uuid = "a33af91c-f02d-484b-be07-31d278c5ca2b" | |
version = "0.1.1" | |
[[deps.Conda]] | |
deps = ["Downloads", "JSON", "VersionParsing"] | |
git-tree-sha1 = "e32a90da027ca45d84678b826fffd3110bb3fc90" | |
uuid = "8f4d0f93-b110-5947-807f-2305c1781a2d" | |
version = "1.8.0" | |
[[deps.ConstructionBase]] | |
deps = ["LinearAlgebra"] | |
git-tree-sha1 = "fb21ddd70a051d882a1686a5a550990bbe371a95" | |
uuid = "187b0558-2788-49d3-abe0-74a17ed4e7c9" | |
version = "1.4.1" | |
[[deps.ContextVariablesX]] | |
deps = ["Compat", "Logging", "UUIDs"] | |
git-tree-sha1 = "25cc3803f1030ab855e383129dcd3dc294e322cc" | |
uuid = "6add18c4-b38d-439d-96f6-d6bc489c04c5" | |
version = "0.1.3" | |
[[deps.Contour]] | |
git-tree-sha1 = "d05d9e7b7aedff4e5b51a029dced05cfb6125781" | |
uuid = "d38c429a-6771-53c6-b99e-75d170b6e991" | |
version = "0.6.2" | |
[[deps.CoordinateTransformations]] | |
deps = ["LinearAlgebra", "StaticArrays"] | |
git-tree-sha1 = "681ea870b918e7cff7111da58791d7f718067a19" | |
uuid = "150eb455-5306-5404-9cee-2592286d6298" | |
version = "0.6.2" | |
[[deps.Crayons]] | |
git-tree-sha1 = "249fe38abf76d48563e2f4556bebd215aa317e15" | |
uuid = "a8cc5b0e-0ffa-5ad4-8c14-923d3ee1735f" | |
version = "4.1.1" | |
[[deps.DLPack]] | |
deps = ["Requires"] | |
git-tree-sha1 = "36489421cca18b9a032c351012043bda78933b2f" | |
uuid = "53c2dc0f-f7d5-43fd-8906-6c0220547083" | |
version = "0.1.2" | |
[[deps.DataAPI]] | |
git-tree-sha1 = "e8119c1a33d267e16108be441a287a6981ba1630" | |
uuid = "9a962f9c-6df0-11e9-0e5d-c546b8b5ee8a" | |
version = "1.14.0" | |
[[deps.DataAugmentation]] | |
deps = ["ColorBlendModes", "CoordinateTransformations", "Distributions", "ImageCore", "ImageDraw", "ImageTransformations", "IndirectArrays", "Interpolations", "LinearAlgebra", "MosaicViews", "OffsetArrays", "Parameters", "Random", "Rotations", "Setfield", "StaticArrays", "Statistics", "Test"] | |
git-tree-sha1 = "9073179282095c1ce9590e19525ad864a7b83211" | |
uuid = "88a5189c-e7ff-4f85-ac6b-e6158070f02e" | |
version = "0.2.11" | |
[[deps.DataDeps]] | |
deps = ["HTTP", "Libdl", "Reexport", "SHA", "p7zip_jll"] | |
git-tree-sha1 = "bc0a264d3e7b3eeb0b6fc9f6481f970697f29805" | |
uuid = "124859b0-ceae-595e-8997-d05f6a7a8dfe" | |
version = "0.7.10" | |
[[deps.DataFrames]] | |
deps = ["Compat", "DataAPI", "Future", "InvertedIndices", "IteratorInterfaceExtensions", "LinearAlgebra", "Markdown", "Missings", "PooledArrays", "PrettyTables", "Printf", "REPL", "Reexport", "SortingAlgorithms", "Statistics", "TableTraits", "Tables", "Unicode"] | |
git-tree-sha1 = "db2a9cb664fcea7836da4b414c3278d71dd602d2" | |
uuid = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0" | |
version = "1.3.6" | |
[[deps.DataStructures]] | |
deps = ["Compat", "InteractiveUtils", "OrderedCollections"] | |
git-tree-sha1 = "d1fff3a548102f48987a52a2e0d114fa97d730f0" | |
uuid = "864edb3b-99cc-5e75-8d2d-829cb0a9cfe8" | |
version = "0.18.13" | |
[[deps.DataValueInterfaces]] | |
git-tree-sha1 = "bfc1187b79289637fa0ef6d4436ebdfe6905cbd6" | |
uuid = "e2d170a0-9d28-54be-80f0-106bbe20a464" | |
version = "1.0.0" | |
[[deps.Dates]] | |
deps = ["Printf"] | |
uuid = "ade2ca70-3891-5945-98fb-dc099432e06a" | |
[[deps.DefineSingletons]] | |
git-tree-sha1 = "0fba8b706d0178b4dc7fd44a96a92382c9065c2c" | |
uuid = "244e2a9f-e319-4986-a169-4d1fe445cd52" | |
version = "0.1.2" | |
[[deps.DelimitedFiles]] | |
deps = ["Mmap"] | |
git-tree-sha1 = "9e2f36d3c96a820c678f2f1f1782582fcf685bae" | |
uuid = "8bb1440f-4735-579b-a4ab-409b98df4dab" | |
version = "1.9.1" | |
[[deps.DensityInterface]] | |
deps = ["InverseFunctions", "Test"] | |
git-tree-sha1 = "80c3e8639e3353e5d2912fb3a1916b8455e2494b" | |
uuid = "b429d917-457f-4dbc-8f4c-0cc954292b1d" | |
version = "0.4.0" | |
[[deps.DiffResults]] | |
deps = ["StaticArraysCore"] | |
git-tree-sha1 = "782dd5f4561f5d267313f23853baaaa4c52ea621" | |
uuid = "163ba53b-c6d8-5494-b064-1a9d43ac40c5" | |
version = "1.1.0" | |
[[deps.DiffRules]] | |
deps = ["IrrationalConstants", "LogExpFunctions", "NaNMath", "Random", "SpecialFunctions"] | |
git-tree-sha1 = "c5b6685d53f933c11404a3ae9822afe30d522494" | |
uuid = "b552c78f-8df3-52c6-915a-8e097449b14b" | |
version = "1.12.2" | |
[[deps.Distances]] | |
deps = ["LinearAlgebra", "SparseArrays", "Statistics", "StatsAPI"] | |
git-tree-sha1 = "3258d0659f812acde79e8a74b11f17ac06d0ca04" | |
uuid = "b4f34e82-e78d-54a5-968a-f98e89d6e8f7" | |
version = "0.10.7" | |
[[deps.Distributed]] | |
deps = ["Random", "Serialization", "Sockets"] | |
uuid = "8ba89e20-285c-5b6f-9357-94700520ee1b" | |
[[deps.Distributions]] | |
deps = ["ChainRulesCore", "DensityInterface", "FillArrays", "LinearAlgebra", "PDMats", "Printf", "QuadGK", "Random", "SparseArrays", "SpecialFunctions", "Statistics", "StatsBase", "StatsFuns", "Test"] | |
git-tree-sha1 = "74911ad88921455c6afcad1eefa12bd7b1724631" | |
uuid = "31c24e10-a181-5473-b8eb-7969acd0382f" | |
version = "0.25.80" | |
[[deps.DocStringExtensions]] | |
deps = ["LibGit2"] | |
git-tree-sha1 = "2fb1e02f2b635d0845df5d7c167fec4dd739b00d" | |
uuid = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae" | |
version = "0.9.3" | |
[[deps.Downloads]] | |
deps = ["ArgTools", "FileWatching", "LibCURL", "NetworkOptions"] | |
uuid = "f43a241f-c20a-4ad4-852c-f6b1247861c6" | |
version = "1.6.0" | |
[[deps.DualNumbers]] | |
deps = ["Calculus", "NaNMath", "SpecialFunctions"] | |
git-tree-sha1 = "5837a837389fccf076445fce071c8ddaea35a566" | |
uuid = "fa6b7ba4-c1ee-5f82-b5fc-ecf0adba8f74" | |
version = "0.6.8" | |
[[deps.EarlyStopping]] | |
deps = ["Dates", "Statistics"] | |
git-tree-sha1 = "98fdf08b707aaf69f524a6cd0a67858cefe0cfb6" | |
uuid = "792122b4-ca99-40de-a6bc-6742525f08b6" | |
version = "0.3.0" | |
[[deps.ExprTools]] | |
git-tree-sha1 = "56559bbef6ca5ea0c0818fa5c90320398a6fbf8d" | |
uuid = "e2ba6199-217a-4e67-a87a-7c52f15ade04" | |
version = "0.1.8" | |
[[deps.FLoops]] | |
deps = ["BangBang", "Compat", "FLoopsBase", "InitialValues", "JuliaVariables", "MLStyle", "Serialization", "Setfield", "Transducers"] | |
git-tree-sha1 = "ffb97765602e3cbe59a0589d237bf07f245a8576" | |
uuid = "cc61a311-1640-44b5-9fba-1b764f453329" | |
version = "0.2.1" | |
[[deps.FLoopsBase]] | |
deps = ["ContextVariablesX"] | |
git-tree-sha1 = "656f7a6859be8673bf1f35da5670246b923964f7" | |
uuid = "b9860ae5-e623-471e-878b-f6a53c775ea6" | |
version = "0.1.1" | |
[[deps.FastAI]] | |
deps = ["DataDeps", "FeatureRegistries", "FileIO", "FilePathsBase", "Flux", "FluxTraining", "InlineTest", "JLD2", "MLDatasets", "MLUtils", "Markdown", "Parameters", "PrettyTables", "Reexport", "Requires", "Setfield", "Statistics", "Test", "TextAnalysis", "UnicodePlots"] | |
git-tree-sha1 = "90b6378e5a53e1e700cb7ccb8ae8fe2b00014034" | |
repo-rev = "lo/metalhead-models" | |
repo-url = "/home/romeo/Documents/github/FastAI.jl" | |
uuid = "5d0beca9-ade8-49ae-ad0b-a3cf890e669f" | |
version = "0.5.0" | |
[[deps.FastTimm]] | |
deps = ["FastAI", "FastVision", "Metalhead", "PyCall", "PyNNTraining"] | |
git-tree-sha1 = "d2b6712b57de0da5177d2a6bdece010d1bee816e" | |
repo-rev = "main" | |
repo-url = "git@github.com:lorenzoh/FastTimm.jl.git" | |
uuid = "6fe279a3-76fd-4f59-807e-d8e68e6f60cf" | |
version = "0.1.0" | |
[[deps.FastVision]] | |
deps = ["ColorVectorSpace", "Colors", "DataAugmentation", "DataDeps", "FastAI", "FixedPointNumbers", "Flux", "ImageIO", "ImageInTerminal", "IndirectArrays", "InlineTest", "MLUtils", "MakieCore", "Metalhead", "ProgressMeter", "ShowCases", "StaticArrays", "Statistics", "UnicodePlots", "Zygote"] | |
path = "/home/romeo/Documents/github/FastAI.jl/FastVision" | |
uuid = "7bf02486-ff4c-4e73-b158-40c00866b54f" | |
version = "0.1.0" | |
[[deps.FeatureRegistries]] | |
deps = ["Base64", "ImageShow", "InlineTest", "Markdown", "PrettyTables", "Setfield", "StructArrays"] | |
git-tree-sha1 = "1be57ce6d4a7cb8dcd9bf2fdb6fbb5e3654c698b" | |
uuid = "c6aefb4f-3ac3-4095-8805-528476b02c02" | |
version = "0.1.2" | |
[[deps.FileIO]] | |
deps = ["Pkg", "Requires", "UUIDs"] | |
git-tree-sha1 = "7be5f99f7d15578798f338f5433b6c432ea8037b" | |
uuid = "5789e2e9-d7fb-5bc7-8068-2c6fae9b9549" | |
version = "1.16.0" | |
[[deps.FilePathsBase]] | |
deps = ["Compat", "Dates", "Mmap", "Printf", "Test", "UUIDs"] | |
git-tree-sha1 = "e27c4ebe80e8699540f2d6c805cc12203b614f12" | |
uuid = "48062228-2e41-5def-b9a4-89aafe57970f" | |
version = "0.9.20" | |
[[deps.FileWatching]] | |
uuid = "7b1f6079-737a-58dc-b8bc-7a2ca5c1b5ee" | |
[[deps.FillArrays]] | |
deps = ["LinearAlgebra", "Random", "SparseArrays", "Statistics"] | |
git-tree-sha1 = "d3ba08ab64bdfd27234d3f61956c966266757fe6" | |
uuid = "1a297f60-69ca-5386-bcde-b61e274b549b" | |
version = "0.13.7" | |
[[deps.FixedPointNumbers]] | |
deps = ["Statistics"] | |
git-tree-sha1 = "335bfdceacc84c5cdf16aadc768aa5ddfc5383cc" | |
uuid = "53c48c17-4a7d-5ca2-90c5-79b7896eea93" | |
version = "0.8.4" | |
[[deps.Flux]] | |
deps = ["Adapt", "CUDA", "ChainRulesCore", "Functors", "LinearAlgebra", "MLUtils", "MacroTools", "NNlib", "NNlibCUDA", "OneHotArrays", "Optimisers", "ProgressLogging", "Random", "Reexport", "SparseArrays", "SpecialFunctions", "Statistics", "StatsBase", "Test", "Zygote"] | |
git-tree-sha1 = "fdbac308c552f749122b0ad8d7cdca02ac131531" | |
uuid = "587475ba-b771-5e3f-ad9e-33799f191a9c" | |
version = "0.13.8" | |
[[deps.FluxTraining]] | |
deps = ["BSON", "ChainRulesCore", "DataStructures", "EarlyStopping", "Flux", "Glob", "Graphs", "ImageCore", "InlineTest", "OnlineStats", "Optimisers", "ParameterSchedulers", "Parameters", "PrettyTables", "ProgressMeter", "Reexport", "Setfield", "SnoopPrecompile", "Statistics", "TensorBoardLogger", "UUIDs", "ValueHistories", "Zygote"] | |
git-tree-sha1 = "758ae8fdb3c3e7b27515a75447db5b194d73ac7c" | |
uuid = "7bf95e4d-ca32-48da-9824-f0dc5310474f" | |
version = "0.3.5" | |
[[deps.FoldsThreads]] | |
deps = ["Accessors", "FunctionWrappers", "InitialValues", "SplittablesBase", "Transducers"] | |
git-tree-sha1 = "eb8e1989b9028f7e0985b4268dabe94682249025" | |
uuid = "9c68100b-dfe1-47cf-94c8-95104e173443" | |
version = "0.1.1" | |
[[deps.Formatting]] | |
deps = ["Printf"] | |
git-tree-sha1 = "8339d61043228fdd3eb658d86c926cb282ae72a8" | |
uuid = "59287772-0a20-5a39-b81b-1366585eb4c0" | |
version = "0.4.2" | |
[[deps.ForwardDiff]] | |
deps = ["CommonSubexpressions", "DiffResults", "DiffRules", "LinearAlgebra", "LogExpFunctions", "NaNMath", "Preferences", "Printf", "Random", "SpecialFunctions", "StaticArrays"] | |
git-tree-sha1 = "a69dd6db8a809f78846ff259298678f0d6212180" | |
uuid = "f6369f11-7733-5829-9624-2563aa707210" | |
version = "0.10.34" | |
[[deps.FreeType]] | |
deps = ["CEnum", "FreeType2_jll"] | |
git-tree-sha1 = "cabd77ab6a6fdff49bfd24af2ebe76e6e018a2b4" | |
uuid = "b38be410-82b0-50bf-ab77-7b57e271db43" | |
version = "4.0.0" | |
[[deps.FreeType2_jll]] | |
deps = ["Artifacts", "Bzip2_jll", "JLLWrappers", "Libdl", "Pkg", "Zlib_jll"] | |
git-tree-sha1 = "87eb71354d8ec1a96d4a7636bd57a7347dde3ef9" | |
uuid = "d7e528f0-a631-5988-bf34-fe36492bcfd7" | |
version = "2.10.4+0" | |
[[deps.FunctionWrappers]] | |
git-tree-sha1 = "d62485945ce5ae9c0c48f124a84998d755bae00e" | |
uuid = "069b7b12-0de2-55c6-9aab-29f3d0a68a2e" | |
version = "1.1.3" | |
[[deps.Functors]] | |
deps = ["LinearAlgebra"] | |
git-tree-sha1 = "a2657dd0f3e8a61dbe70fc7c122038bd33790af5" | |
uuid = "d9f16b24-f501-4c13-a1f2-28368ffc5196" | |
version = "0.3.0" | |
[[deps.Future]] | |
deps = ["Random"] | |
uuid = "9fa8497b-333b-5362-9e8d-4d0656e87820" | |
[[deps.GPUArrays]] | |
deps = ["Adapt", "GPUArraysCore", "LLVM", "LinearAlgebra", "Printf", "Random", "Reexport", "Serialization", "Statistics"] | |
git-tree-sha1 = "4dfaff044eb2ce11a897fecd85538310e60b91e6" | |
uuid = "0c68f7d7-f131-5f86-a1c3-88cf8149b2d7" | |
version = "8.6.2" | |
[[deps.GPUArraysCore]] | |
deps = ["Adapt"] | |
git-tree-sha1 = "57f7cde02d7a53c9d1d28443b9f11ac5fbe7ebc9" | |
uuid = "46192b85-c4d5-4398-a991-12ede77f4527" | |
version = "0.1.3" | |
[[deps.GPUCompiler]] | |
deps = ["ExprTools", "InteractiveUtils", "LLVM", "Libdl", "Logging", "TimerOutputs", "UUIDs"] | |
git-tree-sha1 = "95185985a5d2388c6d0fedb06181ad4ddd40e0cb" | |
uuid = "61eb1bfa-7361-4325-ad38-22787b887f55" | |
version = "0.17.2" | |
[[deps.GZip]] | |
deps = ["Libdl"] | |
git-tree-sha1 = "039be665faf0b8ae36e089cd694233f5dee3f7d6" | |
uuid = "92fee26a-97fe-5a0c-ad85-20a5f3185b63" | |
version = "0.5.1" | |
[[deps.Glob]] | |
git-tree-sha1 = "4df9f7e06108728ebf00a0a11edee4b29a482bb2" | |
uuid = "c27321d9-0574-5035-807b-f59d2c89b15c" | |
version = "1.3.0" | |
[[deps.Graphics]] | |
deps = ["Colors", "LinearAlgebra", "NaNMath"] | |
git-tree-sha1 = "d61890399bc535850c4bf08e4e0d3a7ad0f21cbd" | |
uuid = "a2bd30eb-e257-5431-a919-1863eab51364" | |
version = "1.1.2" | |
[[deps.Graphs]] | |
deps = ["ArnoldiMethod", "Compat", "DataStructures", "Distributed", "Inflate", "LinearAlgebra", "Random", "SharedArrays", "SimpleTraits", "SparseArrays", "Statistics"] | |
git-tree-sha1 = "ba2d094a88b6b287bd25cfa86f301e7693ffae2f" | |
uuid = "86223c79-3864-5bf0-83f7-82e725a168b6" | |
version = "1.7.4" | |
[[deps.HDF5]] | |
deps = ["Compat", "HDF5_jll", "Libdl", "Mmap", "Random", "Requires", "UUIDs"] | |
git-tree-sha1 = "b5df7c3cab3a00c33c2e09c6bd23982a75e2fbb2" | |
uuid = "f67ccb44-e63f-5c2f-98bd-6dc0ccc4ba2f" | |
version = "0.16.13" | |
[[deps.HDF5_jll]] | |
deps = ["Artifacts", "JLLWrappers", "LibCURL_jll", "Libdl", "OpenSSL_jll", "Pkg", "Zlib_jll"] | |
git-tree-sha1 = "4cc2bb72df6ff40b055295fdef6d92955f9dede8" | |
uuid = "0234f1f7-429e-5d53-9886-15a909be8d59" | |
version = "1.12.2+2" | |
[[deps.HTML_Entities]] | |
deps = ["StrTables"] | |
git-tree-sha1 = "c4144ed3bc5f67f595622ad03c0e39fa6c70ccc7" | |
uuid = "7693890a-d069-55fe-a829-b4a6d304f0ee" | |
version = "1.0.1" | |
[[deps.HTTP]] | |
deps = ["Base64", "CodecZlib", "Dates", "IniFile", "Logging", "LoggingExtras", "MbedTLS", "NetworkOptions", "OpenSSL", "Random", "SimpleBufferStream", "Sockets", "URIs", "UUIDs"] | |
git-tree-sha1 = "37e4657cd56b11abe3d10cd4a1ec5fbdb4180263" | |
uuid = "cd3eb016-35fb-5094-929b-558a96fad6f3" | |
version = "1.7.4" | |
[[deps.HypergeometricFunctions]] | |
deps = ["DualNumbers", "LinearAlgebra", "OpenLibm_jll", "SpecialFunctions", "Test"] | |
git-tree-sha1 = "709d864e3ed6e3545230601f94e11ebc65994641" | |
uuid = "34004b35-14d8-5ef3-9330-4cdb6864b03a" | |
version = "0.3.11" | |
[[deps.IRTools]] | |
deps = ["InteractiveUtils", "MacroTools", "Test"] | |
git-tree-sha1 = "2e99184fca5eb6f075944b04c22edec29beb4778" | |
uuid = "7869d1d1-7146-5819-86e3-90919afe41df" | |
version = "0.4.7" | |
[[deps.ImageAxes]] | |
deps = ["AxisArrays", "ImageBase", "ImageCore", "Reexport", "SimpleTraits"] | |
git-tree-sha1 = "c54b581a83008dc7f292e205f4c409ab5caa0f04" | |
uuid = "2803e5a7-5153-5ecf-9a86-9b4c37f5f5ac" | |
version = "0.6.10" | |
[[deps.ImageBase]] | |
deps = ["ImageCore", "Reexport"] | |
git-tree-sha1 = "b51bb8cae22c66d0f6357e3bcb6363145ef20835" | |
uuid = "c817782e-172a-44cc-b673-b171935fbb9e" | |
version = "0.1.5" | |
[[deps.ImageCore]] | |
deps = ["AbstractFFTs", "ColorVectorSpace", "Colors", "FixedPointNumbers", "Graphics", "MappedArrays", "MosaicViews", "OffsetArrays", "PaddedViews", "Reexport"] | |
git-tree-sha1 = "acf614720ef026d38400b3817614c45882d75500" | |
uuid = "a09fc81d-aa75-5fe9-8630-4744c3626534" | |
version = "0.9.4" | |
[[deps.ImageDraw]] | |
deps = ["Distances", "ImageCore", "LinearAlgebra"] | |
git-tree-sha1 = "6ed6e945d909f87c3013e391dcd3b2a56e48b331" | |
uuid = "4381153b-2b60-58ae-a1ba-fd683676385f" | |
version = "0.2.5" | |
[[deps.ImageIO]] | |
deps = ["FileIO", "IndirectArrays", "JpegTurbo", "LazyModules", "Netpbm", "OpenEXR", "PNGFiles", "QOI", "Sixel", "TiffImages", "UUIDs"] | |
git-tree-sha1 = "342f789fd041a55166764c351da1710db97ce0e0" | |
uuid = "82e4d734-157c-48bb-816b-45c225c6df19" | |
version = "0.6.6" | |
[[deps.ImageInTerminal]] | |
deps = ["ColorTypes", "Crayons", "FileIO", "Sixel", "XTermColors"] | |
git-tree-sha1 = "f260604e7600723a323b42cb92ae22d837cd5dc9" | |
uuid = "d8c32880-2388-543b-8c61-d9f865259254" | |
version = "0.5.2" | |
[[deps.ImageMetadata]] | |
deps = ["AxisArrays", "ImageAxes", "ImageBase", "ImageCore"] | |
git-tree-sha1 = "36cbaebed194b292590cba2593da27b34763804a" | |
uuid = "bc367c6b-8a6b-528e-b4bd-a4b897500b49" | |
version = "0.9.8" | |
[[deps.ImageShow]] | |
deps = ["Base64", "FileIO", "ImageBase", "ImageCore", "OffsetArrays", "StackViews"] | |
git-tree-sha1 = "b563cf9ae75a635592fc73d3eb78b86220e55bd8" | |
uuid = "4e3cecfd-b093-5904-9786-8bbb286a6a31" | |
version = "0.3.6" | |
[[deps.ImageTransformations]] | |
deps = ["AxisAlgorithms", "ColorVectorSpace", "CoordinateTransformations", "ImageBase", "ImageCore", "Interpolations", "OffsetArrays", "Rotations", "StaticArrays"] | |
git-tree-sha1 = "8717482f4a2108c9358e5c3ca903d3a6113badc9" | |
uuid = "02fcd773-0e25-5acc-982a-7f6622650795" | |
version = "0.9.5" | |
[[deps.Imath_jll]] | |
deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg"] | |
git-tree-sha1 = "87f7662e03a649cffa2e05bf19c303e168732d3e" | |
uuid = "905a6f67-0a94-5f89-b386-d35d92009cd1" | |
version = "3.1.2+0" | |
[[deps.IndirectArrays]] | |
git-tree-sha1 = "012e604e1c7458645cb8b436f8fba789a51b257f" | |
uuid = "9b13fd28-a010-5f03-acff-a1bbcff69959" | |
version = "1.0.0" | |
[[deps.InfiniteArrays]] | |
deps = ["ArrayLayouts", "FillArrays", "Infinities", "LazyArrays", "LinearAlgebra", "Statistics"] | |
git-tree-sha1 = "ed50f2d5dfd7a4cc939431e2f4aa06cf80f735e3" | |
uuid = "4858937d-0d70-526a-a4dd-2d5cb5dd786c" | |
version = "0.12.10" | |
[[deps.Infinities]] | |
git-tree-sha1 = "7cafb0d59dabad36e451c80b03505ecd599ce924" | |
uuid = "e1ba4f0e-776d-440f-acd9-e1d2e9742647" | |
version = "0.1.6" | |
[[deps.Inflate]] | |
git-tree-sha1 = "5cd07aab533df5170988219191dfad0519391428" | |
uuid = "d25df0c9-e2be-5dd7-82c8-3ad0b3e990b9" | |
version = "0.1.3" | |
[[deps.IniFile]] | |
git-tree-sha1 = "f550e6e32074c939295eb5ea6de31849ac2c9625" | |
uuid = "83e8ac13-25f8-5344-8a64-a9f2b223428f" | |
version = "0.5.1" | |
[[deps.InitialValues]] | |
git-tree-sha1 = "4da0f88e9a39111c2fa3add390ab15f3a44f3ca3" | |
uuid = "22cec73e-a1b8-11e9-2c92-598750a2cf9c" | |
version = "0.3.1" | |
[[deps.InlineStrings]] | |
deps = ["Parsers"] | |
git-tree-sha1 = "9cc2baf75c6d09f9da536ddf58eb2f29dedaf461" | |
uuid = "842dd82b-1e85-43dc-bf29-5d0ee9dffc48" | |
version = "1.4.0" | |
[[deps.InlineTest]] | |
deps = ["Test"] | |
git-tree-sha1 = "daf0743879904f0ad645ca6594e1479685f158a2" | |
uuid = "bd334432-b1e7-49c7-a2dc-dd9149e4ebd6" | |
version = "0.2.0" | |
[[deps.InteractiveUtils]] | |
deps = ["Markdown"] | |
uuid = "b77e0a4c-d291-57a0-90e8-8db25a27a240" | |
[[deps.InternedStrings]] | |
deps = ["Random", "Test"] | |
git-tree-sha1 = "eb05b5625bc5d821b8075a77e4c421933e20c76b" | |
uuid = "7d512f48-7fb1-5a58-b986-67e6dc259f01" | |
version = "0.7.0" | |
[[deps.Interpolations]] | |
deps = ["AxisAlgorithms", "ChainRulesCore", "LinearAlgebra", "OffsetArrays", "Random", "Ratios", "Requires", "SharedArrays", "SparseArrays", "StaticArrays", "WoodburyMatrices"] | |
git-tree-sha1 = "b7bc05649af456efc75d178846f47006c2c4c3c7" | |
uuid = "a98d9a8b-a2ab-59e6-89dd-64a1c18fca59" | |
version = "0.13.6" | |
[[deps.IntervalSets]] | |
deps = ["Dates", "Random", "Statistics"] | |
git-tree-sha1 = "16c0cc91853084cb5f58a78bd209513900206ce6" | |
uuid = "8197267c-284f-5f27-9208-e0e47529a953" | |
version = "0.7.4" | |
[[deps.InverseFunctions]] | |
deps = ["Test"] | |
git-tree-sha1 = "49510dfcb407e572524ba94aeae2fced1f3feb0f" | |
uuid = "3587e190-3f89-42d0-90ee-14403ec27112" | |
version = "0.1.8" | |
[[deps.InvertedIndices]] | |
git-tree-sha1 = "82aec7a3dd64f4d9584659dc0b62ef7db2ef3e19" | |
uuid = "41ab1584-1d38-5bbf-9106-f11c6c58b48f" | |
version = "1.2.0" | |
[[deps.IrrationalConstants]] | |
git-tree-sha1 = "7fd44fd4ff43fc60815f8e764c0f352b83c49151" | |
uuid = "92d709cd-6900-40b7-9082-c6be49f344b6" | |
version = "0.1.1" | |
[[deps.IterTools]] | |
git-tree-sha1 = "fa6287a4469f5e048d763df38279ee729fbd44e5" | |
uuid = "c8e1da08-722c-5040-9ed9-7db0dc04731e" | |
version = "1.4.0" | |
[[deps.IteratorInterfaceExtensions]] | |
git-tree-sha1 = "a3f24677c21f5bbe9d2a714f95dcd58337fb2856" | |
uuid = "82899510-4779-5014-852e-03e436cf321d" | |
version = "1.0.0" | |
[[deps.JLD2]] | |
deps = ["FileIO", "MacroTools", "Mmap", "OrderedCollections", "Pkg", "Printf", "Reexport", "TranscodingStreams", "UUIDs"] | |
git-tree-sha1 = "c3244ef42b7d4508c638339df1bdbf4353e144db" | |
uuid = "033835bb-8acc-5ee8-8aae-3f567f8a3819" | |
version = "0.4.30" | |
[[deps.JLLWrappers]] | |
deps = ["Preferences"] | |
git-tree-sha1 = "abc9885a7ca2052a736a600f7fa66209f96506e1" | |
uuid = "692b3bcd-3c85-4b1f-b108-f13ce0eb3210" | |
version = "1.4.1" | |
[[deps.JSON]] | |
deps = ["Dates", "Mmap", "Parsers", "Unicode"] | |
git-tree-sha1 = "3c837543ddb02250ef42f4738347454f95079d4e" | |
uuid = "682c06a0-de6a-54ab-a142-c8b1cf79cde6" | |
version = "0.21.3" | |
[[deps.JSON3]] | |
deps = ["Dates", "Mmap", "Parsers", "SnoopPrecompile", "StructTypes", "UUIDs"] | |
git-tree-sha1 = "84b10656a41ef564c39d2d477d7236966d2b5683" | |
uuid = "0f8b85d8-7281-11e9-16c2-39a750bddbf1" | |
version = "1.12.0" | |
[[deps.JpegTurbo]] | |
deps = ["CEnum", "FileIO", "ImageCore", "JpegTurbo_jll", "TOML"] | |
git-tree-sha1 = "106b6aa272f294ba47e96bd3acbabdc0407b5c60" | |
uuid = "b835a17e-a41a-41e7-81f0-2f016b05efe0" | |
version = "0.1.2" | |
[[deps.JpegTurbo_jll]] | |
deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg"] | |
git-tree-sha1 = "b53380851c6e6664204efb2e62cd24fa5c47e4ba" | |
uuid = "aacddb02-875f-59d6-b918-886e6ef4fbf8" | |
version = "2.1.2+0" | |
[[deps.JuliaVariables]] | |
deps = ["MLStyle", "NameResolution"] | |
git-tree-sha1 = "49fb3cb53362ddadb4415e9b73926d6b40709e70" | |
uuid = "b14d175d-62b4-44ba-8fb7-3064adc8c3ec" | |
version = "0.2.4" | |
[[deps.LLVM]] | |
deps = ["CEnum", "LLVMExtra_jll", "Libdl", "Printf", "Unicode"] | |
git-tree-sha1 = "b8ae281340f0d3e973aae7b96fb7502b0119b376" | |
uuid = "929cbde3-209d-540e-8aea-75f648917ca0" | |
version = "4.15.0" | |
[[deps.LLVMExtra_jll]] | |
deps = ["Artifacts", "JLLWrappers", "LazyArtifacts", "Libdl", "Pkg", "TOML"] | |
git-tree-sha1 = "771bfe376249626d3ca12bcd58ba243d3f961576" | |
uuid = "dad2f222-ce93-54a1-a47d-0025e8a3acab" | |
version = "0.0.16+0" | |
[[deps.Languages]] | |
deps = ["InteractiveUtils", "JSON"] | |
git-tree-sha1 = "b1a564061268ccc3f3397ac0982983a657d4dcb8" | |
uuid = "8ef0a80b-9436-5d2c-a485-80b904378c43" | |
version = "0.4.3" | |
[[deps.LazyArrays]] | |
deps = ["ArrayLayouts", "FillArrays", "LinearAlgebra", "MacroTools", "MatrixFactorizations", "SparseArrays", "StaticArrays"] | |
git-tree-sha1 = "f6742270b551e03bd8ba64e1c754f5fab089ab63" | |
uuid = "5078a376-72f3-5289-bfd5-ec5146d43c02" | |
version = "0.22.16" | |
[[deps.LazyArtifacts]] | |
deps = ["Artifacts", "Pkg"] | |
uuid = "4af54fe1-eca0-43a8-85a7-787d91b784e3" | |
[[deps.LazyModules]] | |
git-tree-sha1 = "a560dd966b386ac9ae60bdd3a3d3a326062d3c3e" | |
uuid = "8cdb02fc-e678-4876-92c5-9defec4f444e" | |
version = "0.3.1" | |
[[deps.LibCURL]] | |
deps = ["LibCURL_jll", "MozillaCACerts_jll"] | |
uuid = "b27032c2-a3e7-50c8-80cd-2d36dbcbfd21" | |
version = "0.6.3" | |
[[deps.LibCURL_jll]] | |
deps = ["Artifacts", "LibSSH2_jll", "Libdl", "MbedTLS_jll", "Zlib_jll", "nghttp2_jll"] | |
uuid = "deac9b47-8bc7-5906-a0fe-35ac56dc84c0" | |
version = "7.84.0+0" | |
[[deps.LibGit2]] | |
deps = ["Base64", "NetworkOptions", "Printf", "SHA"] | |
uuid = "76f85450-5226-5b5a-8eaa-529ad045b433" | |
[[deps.LibSSH2_jll]] | |
deps = ["Artifacts", "Libdl", "MbedTLS_jll"] | |
uuid = "29816b5a-b9ab-546f-933c-edad1886dfa8" | |
version = "1.10.2+0" | |
[[deps.Libdl]] | |
uuid = "8f399da3-3557-5675-b5ff-fb832c97cbdb" | |
[[deps.Libiconv_jll]] | |
deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg"] | |
git-tree-sha1 = "c7cb1f5d892775ba13767a87c7ada0b980ea0a71" | |
uuid = "94ce4f54-9a6c-5748-9c1c-f9c7231a4531" | |
version = "1.16.1+2" | |
[[deps.LinearAlgebra]] | |
deps = ["Libdl", "OpenBLAS_jll", "libblastrampoline_jll"] | |
uuid = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e" | |
[[deps.LogExpFunctions]] | |
deps = ["DocStringExtensions", "IrrationalConstants", "LinearAlgebra"] | |
git-tree-sha1 = "680e733c3a0a9cea9e935c8c2184aea6a63fa0b5" | |
uuid = "2ab3a3ac-af41-5b50-aa03-7779005ae688" | |
version = "0.3.21" | |
[deps.LogExpFunctions.extensions] | |
ChainRulesCoreExt = "ChainRulesCore" | |
ChangesOfVariablesExt = "ChangesOfVariables" | |
InverseFunctionsExt = "InverseFunctions" | |
[deps.LogExpFunctions.weakdeps] | |
ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4" | |
ChangesOfVariables = "9e997f8a-9a97-42d5-a9f1-ce6bfc15e2c0" | |
InverseFunctions = "3587e190-3f89-42d0-90ee-14403ec27112" | |
[[deps.Logging]] | |
uuid = "56ddb016-857b-54e1-b83d-db4d58db5568" | |
[[deps.LoggingExtras]] | |
deps = ["Dates", "Logging"] | |
git-tree-sha1 = "cedb76b37bc5a6c702ade66be44f831fa23c681e" | |
uuid = "e6f89c97-d47a-5376-807f-9c37f3926c36" | |
version = "1.0.0" | |
[[deps.MAT]] | |
deps = ["BufferedStreams", "CodecZlib", "HDF5", "SparseArrays"] | |
git-tree-sha1 = "971be550166fe3f604d28715302b58a3f7293160" | |
uuid = "23992714-dd62-5051-b70f-ba57cb901cac" | |
version = "0.10.3" | |
[[deps.MLDatasets]] | |
deps = ["CSV", "Chemfiles", "DataDeps", "DataFrames", "DelimitedFiles", "FileIO", "FixedPointNumbers", "GZip", "Glob", "HDF5", "ImageShow", "JLD2", "JSON3", "LazyModules", "MAT", "MLUtils", "NPZ", "Pickle", "Printf", "Requires", "SparseArrays", "Tables"] | |
git-tree-sha1 = "498b37aa3ebb4407adea36df1b244fa4e397de5e" | |
uuid = "eb30cadb-4394-5ae3-aed4-317e484a6458" | |
version = "0.7.9" | |
[[deps.MLStyle]] | |
git-tree-sha1 = "060ef7956fef2dc06b0e63b294f7dbfbcbdc7ea2" | |
uuid = "d8e11817-5142-5d16-987a-aa16d5891078" | |
version = "0.4.16" | |
[[deps.MLUtils]] | |
deps = ["ChainRulesCore", "DataAPI", "DelimitedFiles", "FLoops", "FoldsThreads", "NNlib", "Random", "ShowCases", "SimpleTraits", "Statistics", "StatsBase", "Tables", "Transducers"] | |
git-tree-sha1 = "82c1104919d664ab1024663ad851701415300c5f" | |
uuid = "f1d291b0-491e-4a28-83b9-f70985020b54" | |
version = "0.3.1" | |
[[deps.MacroTools]] | |
deps = ["Markdown", "Random"] | |
git-tree-sha1 = "42324d08725e200c23d4dfb549e0d5d89dede2d2" | |
uuid = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09" | |
version = "0.5.10" | |
[[deps.MakieCore]] | |
deps = ["Observables"] | |
git-tree-sha1 = "2c3fc86d52dfbada1a2e5e150e50f06c30ef149c" | |
uuid = "20f20a25-4f0e-4fdf-b5d1-57303727442b" | |
version = "0.6.2" | |
[[deps.MappedArrays]] | |
git-tree-sha1 = "e8b359ef06ec72e8c030463fe02efe5527ee5142" | |
uuid = "dbb5928d-eab1-5f90-85c2-b9b0edb7c900" | |
version = "0.4.1" | |
[[deps.MarchingCubes]] | |
deps = ["SnoopPrecompile", "StaticArrays"] | |
git-tree-sha1 = "55aaf3fdf414b691a15875cfe5edb6e0daf4625a" | |
uuid = "299715c1-40a9-479a-aaf9-4a633d36f717" | |
version = "0.1.6" | |
[[deps.Markdown]] | |
deps = ["Base64"] | |
uuid = "d6f4376e-aef5-505a-96c1-9c027394607a" | |
[[deps.MatrixFactorizations]] | |
deps = ["ArrayLayouts", "LinearAlgebra", "Printf", "Random"] | |
git-tree-sha1 = "22cf8c4315c417300b753c0def0fcb94e83d79b1" | |
uuid = "a3b82374-2e81-5b9e-98ce-41277c0e4c87" | |
version = "0.9.5" | |
[[deps.MbedTLS]] | |
deps = ["Dates", "MbedTLS_jll", "MozillaCACerts_jll", "Random", "Sockets"] | |
git-tree-sha1 = "03a9b9718f5682ecb107ac9f7308991db4ce395b" | |
uuid = "739be429-bea8-5141-9913-cc70e7f3736d" | |
version = "1.1.7" | |
[[deps.MbedTLS_jll]] | |
deps = ["Artifacts", "Libdl"] | |
uuid = "c8ffd9c3-330d-5841-b78e-0817d7145fa1" | |
version = "2.28.0+0" | |
[[deps.Metalhead]] | |
deps = ["Artifacts", "BSON", "CUDA", "ChainRulesCore", "Flux", "Functors", "LazyArtifacts", "MLUtils", "NNlib", "NNlibCUDA", "PartialFunctions", "Random", "Statistics"] | |
git-tree-sha1 = "f1744a1d5e3dfe5fbc7e552e5e642e413d35def9" | |
repo-rev = "master" | |
repo-url = "https://github.com/FluxML/Metalhead.jl.git" | |
uuid = "dbeba491-748d-5e0e-a39e-b530a07fa0cc" | |
version = "0.8.0-DEV" | |
[[deps.MicroCollections]] | |
deps = ["BangBang", "InitialValues", "Setfield"] | |
git-tree-sha1 = "4d5917a26ca33c66c8e5ca3247bd163624d35493" | |
uuid = "128add7d-3638-4c79-886c-908ea0c25c34" | |
version = "0.1.3" | |
[[deps.Missings]] | |
deps = ["DataAPI"] | |
git-tree-sha1 = "f66bdc5de519e8f8ae43bdc598782d35a25b1272" | |
uuid = "e1d29d7a-bbdc-5cf2-9ac0-f12de2c33e28" | |
version = "1.1.0" | |
[[deps.Mmap]] | |
uuid = "a63ad114-7e13-5084-954f-fe012c677804" | |
[[deps.MosaicViews]] | |
deps = ["MappedArrays", "OffsetArrays", "PaddedViews", "StackViews"] | |
git-tree-sha1 = "7b86a5d4d70a9f5cdf2dacb3cbe6d251d1a61dbe" | |
uuid = "e94cdb99-869f-56ef-bcf0-1ae2bcbe0389" | |
version = "0.3.4" | |
[[deps.MozillaCACerts_jll]] | |
uuid = "14a3606d-f60d-562e-9121-12d972cd8159" | |
version = "2022.10.11" | |
[[deps.NNlib]] | |
deps = ["Adapt", "ChainRulesCore", "LinearAlgebra", "Pkg", "Random", "Requires", "Statistics"] | |
git-tree-sha1 = "e69bf048c892296a45083f0abab02ee8cffe1378" | |
uuid = "872c559c-99b0-510c-b3b7-b6c96a88d5cd" | |
version = "0.8.17" | |
[[deps.NNlibCUDA]] | |
deps = ["Adapt", "CUDA", "LinearAlgebra", "NNlib", "Random", "Statistics"] | |
git-tree-sha1 = "b05a082b08a3af0e5c576883bc6dfb6513e7e478" | |
uuid = "a00861dc-f156-4864-bf3c-e6376f28a68d" | |
version = "0.2.6" | |
[[deps.NPZ]] | |
deps = ["FileIO", "ZipFile"] | |
git-tree-sha1 = "60a8e272fe0c5079363b28b0953831e2dd7b7e6f" | |
uuid = "15e1cf62-19b3-5cfa-8e77-841668bca605" | |
version = "0.4.3" | |
[[deps.NaNMath]] | |
deps = ["OpenLibm_jll"] | |
git-tree-sha1 = "a7c3d1da1189a1c2fe843a3bfa04d18d20eb3211" | |
uuid = "77ba4419-2d1f-58cd-9bb1-8ffee604a2e3" | |
version = "1.0.1" | |
[[deps.NameResolution]] | |
deps = ["PrettyPrint"] | |
git-tree-sha1 = "1a0fa0e9613f46c9b8c11eee38ebb4f590013c5e" | |
uuid = "71a1bf82-56d0-4bbc-8a3c-48b961074391" | |
version = "0.1.5" | |
[[deps.Netpbm]] | |
deps = ["FileIO", "ImageCore", "ImageMetadata"] | |
git-tree-sha1 = "5ae7ca23e13855b3aba94550f26146c01d259267" | |
uuid = "f09324ee-3d7c-5217-9330-fc30815ba969" | |
version = "1.1.0" | |
[[deps.NetworkOptions]] | |
uuid = "ca575930-c2e3-43a9-ace4-1e988b2c1908" | |
version = "1.2.0" | |
[[deps.Observables]] | |
git-tree-sha1 = "6862738f9796b3edc1c09d0890afce4eca9e7e93" | |
uuid = "510215fc-4207-5dde-b226-833fc4488ee2" | |
version = "0.5.4" | |
[[deps.OffsetArrays]] | |
deps = ["Adapt"] | |
git-tree-sha1 = "82d7c9e310fe55aa54996e6f7f94674e2a38fcb4" | |
uuid = "6fe1bfb0-de20-5000-8ca7-80f57d26f881" | |
version = "1.12.9" | |
[[deps.OneHotArrays]] | |
deps = ["Adapt", "ChainRulesCore", "Compat", "GPUArraysCore", "LinearAlgebra", "NNlib"] | |
git-tree-sha1 = "f511fca956ed9e70b80cd3417bb8c2dde4b68644" | |
uuid = "0b1bfda6-eb8a-41d2-88d8-f5af5cad476f" | |
version = "0.2.3" | |
[[deps.OnlineStats]] | |
deps = ["AbstractTrees", "Dates", "LinearAlgebra", "OnlineStatsBase", "OrderedCollections", "Random", "RecipesBase", "Statistics", "StatsBase"] | |
git-tree-sha1 = "e81f7f4179becdb6c365fb5a44042cfe7d50b77a" | |
uuid = "a15396b6-48d5-5d58-9928-6d29437db91e" | |
version = "1.5.14" | |
[[deps.OnlineStatsBase]] | |
deps = ["AbstractTrees", "Dates", "LinearAlgebra", "OrderedCollections", "Statistics", "StatsBase"] | |
git-tree-sha1 = "60e587c99ea261d1c452d2acb1f3a481e772660c" | |
uuid = "925886fa-5bf2-5e8e-b522-a9147a512338" | |
version = "1.5.0" | |
[[deps.OpenBLAS_jll]] | |
deps = ["Artifacts", "CompilerSupportLibraries_jll", "Libdl"] | |
uuid = "4536629a-c528-5b80-bd46-f80d51c5b363" | |
version = "0.3.21+0" | |
[[deps.OpenEXR]] | |
deps = ["Colors", "FileIO", "OpenEXR_jll"] | |
git-tree-sha1 = "327f53360fdb54df7ecd01e96ef1983536d1e633" | |
uuid = "52e1d378-f018-4a11-a4be-720524705ac7" | |
version = "0.3.2" | |
[[deps.OpenEXR_jll]] | |
deps = ["Artifacts", "Imath_jll", "JLLWrappers", "Libdl", "Pkg", "Zlib_jll"] | |
git-tree-sha1 = "923319661e9a22712f24596ce81c54fc0366f304" | |
uuid = "18a262bb-aa17-5467-a713-aee519bc75cb" | |
version = "3.1.1+0" | |
[[deps.OpenLibm_jll]] | |
deps = ["Artifacts", "Libdl"] | |
uuid = "05823500-19ac-5b8b-9628-191a04bc5112" | |
version = "0.8.1+0" | |
[[deps.OpenSSL]] | |
deps = ["BitFlags", "Dates", "MozillaCACerts_jll", "OpenSSL_jll", "Sockets"] | |
git-tree-sha1 = "6503b77492fd7fcb9379bf73cd31035670e3c509" | |
uuid = "4d8831e6-92b7-49fb-bdf8-b643e874388c" | |
version = "1.3.3" | |
[[deps.OpenSSL_jll]] | |
deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg"] | |
git-tree-sha1 = "f6e9dba33f9f2c44e08a020b0caf6903be540004" | |
uuid = "458c3c95-2e84-50aa-8efc-19380b2a3a95" | |
version = "1.1.19+0" | |
[[deps.OpenSpecFun_jll]] | |
deps = ["Artifacts", "CompilerSupportLibraries_jll", "JLLWrappers", "Libdl", "Pkg"] | |
git-tree-sha1 = "13652491f6856acfd2db29360e1bbcd4565d04f1" | |
uuid = "efe28fd5-8261-553b-a9e1-b2916fc3738e" | |
version = "0.5.5+0" | |
[[deps.Optimisers]] | |
deps = ["ChainRulesCore", "Functors", "LinearAlgebra", "Random", "Statistics"] | |
git-tree-sha1 = "e657acef119cc0de2a8c0762666d3b64727b053b" | |
uuid = "3bd65402-5787-11e9-1adc-39752487f4e2" | |
version = "0.2.14" | |
[[deps.OrderedCollections]] | |
git-tree-sha1 = "85f8e6578bf1f9ee0d11e7bb1b1456435479d47c" | |
uuid = "bac558e1-5e72-5ebc-8fee-abe8a469f55d" | |
version = "1.4.1" | |
[[deps.PDMats]] | |
deps = ["LinearAlgebra", "SparseArrays", "SuiteSparse"] | |
git-tree-sha1 = "cf494dca75a69712a72b80bc48f59dcf3dea63ec" | |
uuid = "90014a1f-27ba-587c-ab20-58faa44d9150" | |
version = "0.11.16" | |
[[deps.PNGFiles]] | |
deps = ["Base64", "CEnum", "ImageCore", "IndirectArrays", "OffsetArrays", "libpng_jll"] | |
git-tree-sha1 = "f809158b27eba0c18c269cf2a2be6ed751d3e81d" | |
uuid = "f57f5aa1-a3ce-4bc8-8ab9-96f992907883" | |
version = "0.3.17" | |
[[deps.PaddedViews]] | |
deps = ["OffsetArrays"] | |
git-tree-sha1 = "03a7a85b76381a3d04c7a1656039197e70eda03d" | |
uuid = "5432bcbf-9aad-5242-b902-cca2824c8663" | |
version = "0.5.11" | |
[[deps.ParameterSchedulers]] | |
deps = ["Flux", "InfiniteArrays"] | |
git-tree-sha1 = "554af02023b50b5a683690984886852edb1f8aee" | |
uuid = "d7d3b36b-41b8-4d0d-a2bf-768c6151755e" | |
version = "0.3.4" | |
[[deps.Parameters]] | |
deps = ["OrderedCollections", "UnPack"] | |
git-tree-sha1 = "34c0e9ad262e5f7fc75b10a9952ca7692cfc5fbe" | |
uuid = "d96e819e-fc66-5662-9728-84c9c7592b0a" | |
version = "0.12.3" | |
[[deps.Parsers]] | |
deps = ["Dates", "SnoopPrecompile"] | |
git-tree-sha1 = "151d91d63d8d6c1a5789ecb7de51547e00480f1b" | |
uuid = "69de0a69-1ddd-5017-9359-2bf0b02dc9f0" | |
version = "2.5.4" | |
[[deps.PartialFunctions]] | |
git-tree-sha1 = "b3901ea034cfd8aae57a2fa0dde0b0ea18bad1cb" | |
uuid = "570af359-4316-4cb7-8c74-252c00c2016b" | |
version = "1.1.1" | |
[[deps.Pickle]] | |
deps = ["DataStructures", "InternedStrings", "Serialization", "SparseArrays", "Strided", "StringEncodings", "ZipFile"] | |
git-tree-sha1 = "e6a34eb1dc0c498f0774bbfbbbeff2de101f4235" | |
uuid = "fbb45041-c46e-462f-888f-7c521cafbc2c" | |
version = "0.3.2" | |
[[deps.Pkg]] | |
deps = ["Artifacts", "Dates", "Downloads", "FileWatching", "LibGit2", "Libdl", "Logging", "Markdown", "Printf", "REPL", "Random", "SHA", "Serialization", "TOML", "Tar", "UUIDs", "p7zip_jll"] | |
uuid = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f" | |
version = "1.9.0" | |
[[deps.PkgVersion]] | |
deps = ["Pkg"] | |
git-tree-sha1 = "f6cf8e7944e50901594838951729a1861e668cb8" | |
uuid = "eebad327-c553-4316-9ea0-9fa01ccd7688" | |
version = "0.3.2" | |
[[deps.PooledArrays]] | |
deps = ["DataAPI", "Future"] | |
git-tree-sha1 = "a6062fe4063cdafe78f4a0a81cfffb89721b30e7" | |
uuid = "2dfb63ee-cc39-5dd5-95bd-886bf059d720" | |
version = "1.4.2" | |
[[deps.Preferences]] | |
deps = ["TOML"] | |
git-tree-sha1 = "47e5f437cc0e7ef2ce8406ce1e7e24d44915f88d" | |
uuid = "21216c6a-2e73-6563-6e65-726566657250" | |
version = "1.3.0" | |
[[deps.PrettyPrint]] | |
git-tree-sha1 = "632eb4abab3449ab30c5e1afaa874f0b98b586e4" | |
uuid = "8162dcfd-2161-5ef2-ae6c-7681170c5f98" | |
version = "0.2.0" | |
[[deps.PrettyTables]] | |
deps = ["Crayons", "Formatting", "Markdown", "Reexport", "Tables"] | |
git-tree-sha1 = "dfb54c4e414caa595a1f2ed759b160f5a3ddcba5" | |
uuid = "08abe8d2-0d0c-5749-adfa-8a2ac140af0d" | |
version = "1.3.1" | |
[[deps.Printf]] | |
deps = ["Unicode"] | |
uuid = "de0858da-6303-5e67-8744-51eddeeeb8d7" | |
[[deps.ProgressLogging]] | |
deps = ["Logging", "SHA", "UUIDs"] | |
git-tree-sha1 = "80d919dee55b9c50e8d9e2da5eeafff3fe58b539" | |
uuid = "33c8b6b6-d38a-422a-b730-caa89a2f386c" | |
version = "0.1.4" | |
[[deps.ProgressMeter]] | |
deps = ["Distributed", "Printf"] | |
git-tree-sha1 = "d7a7aef8f8f2d537104f170139553b14dfe39fe9" | |
uuid = "92933f4c-e287-5a05-a399-4b506db050ca" | |
version = "1.7.2" | |
[[deps.ProtoBuf]] | |
deps = ["Logging", "protoc_jll"] | |
git-tree-sha1 = "5bb5eb83db702376abe5fb2ef0739b08da3ab461" | |
uuid = "3349acd9-ac6a-5e09-bcdb-63829b23a429" | |
version = "0.11.5" | |
[[deps.PyCall]] | |
deps = ["Conda", "Dates", "Libdl", "LinearAlgebra", "MacroTools", "Serialization", "VersionParsing"] | |
git-tree-sha1 = "62f417f6ad727987c755549e9cd88c46578da562" | |
uuid = "438e738f-606a-5dbb-bf0a-cddfbfd45ab0" | |
version = "1.95.1" | |
[[deps.PyCallChainRules]] | |
deps = ["Adapt", "ChainRulesCore", "DLPack", "FillArrays", "Functors", "PyCall", "Random", "Requires"] | |
git-tree-sha1 = "6c48e210fcc10f3e4054e4dfdfffa7152e8cb686" | |
uuid = "b12ccfe2-7326-416f-9f4f-cd3183bd9fe8" | |
version = "0.4.0" | |
[[deps.PyNNTraining]] | |
deps = ["CUDA", "Flux", "FluxTraining", "Functors", "InlineTest", "Optimisers", "PyCall", "PyCallChainRules", "SnoopPrecompile"] | |
git-tree-sha1 = "7e0b59adf849cebca4b768dfc3df79f97444c90c" | |
repo-rev = "main" | |
repo-url = "git@github.com:lorenzoh/PyNNTraining.jl.git" | |
uuid = "013e1a4a-88fe-473f-be94-6c6af62e62ad" | |
version = "0.1.0" | |
[[deps.QOI]] | |
deps = ["ColorTypes", "FileIO", "FixedPointNumbers"] | |
git-tree-sha1 = "18e8f4d1426e965c7b532ddd260599e1510d26ce" | |
uuid = "4b34888f-f399-49d4-9bb3-47ed5cae4e65" | |
version = "1.0.0" | |
[[deps.QuadGK]] | |
deps = ["DataStructures", "LinearAlgebra"] | |
git-tree-sha1 = "de191bc385072cc6c7ed3ffdc1caeed3f22c74d4" | |
uuid = "1fd47b50-473d-5c70-9696-f719f8f3bcdc" | |
version = "2.7.0" | |
[[deps.Quaternions]] | |
deps = ["LinearAlgebra", "Random", "RealDot"] | |
git-tree-sha1 = "da095158bdc8eaccb7890f9884048555ab771019" | |
uuid = "94ee1d12-ae83-5a48-8b1c-48b8ff168ae0" | |
version = "0.7.4" | |
[[deps.REPL]] | |
deps = ["InteractiveUtils", "Markdown", "Sockets", "Unicode"] | |
uuid = "3fa0cd96-eef1-5676-8a61-b3b8758bbffb" | |
[[deps.Random]] | |
deps = ["SHA", "Serialization"] | |
uuid = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c" | |
[[deps.Random123]] | |
deps = ["Random", "RandomNumbers"] | |
git-tree-sha1 = "7a1a306b72cfa60634f03a911405f4e64d1b718b" | |
uuid = "74087812-796a-5b5d-8853-05524746bad3" | |
version = "1.6.0" | |
[[deps.RandomNumbers]] | |
deps = ["Random", "Requires"] | |
git-tree-sha1 = "043da614cc7e95c703498a491e2c21f58a2b8111" | |
uuid = "e6cf234a-135c-5ec9-84dd-332b85af5143" | |
version = "1.5.3" | |
[[deps.RangeArrays]] | |
git-tree-sha1 = "b9039e93773ddcfc828f12aadf7115b4b4d225f5" | |
uuid = "b3c3ace0-ae52-54e7-9d0b-2c1406fd6b9d" | |
version = "0.3.2" | |
[[deps.Ratios]] | |
deps = ["Requires"] | |
git-tree-sha1 = "dc84268fe0e3335a62e315a3a7cf2afa7178a734" | |
uuid = "c84ed2f1-dad5-54f0-aa8e-dbefe2724439" | |
version = "0.4.3" | |
[[deps.RealDot]] | |
deps = ["LinearAlgebra"] | |
git-tree-sha1 = "9f0a1b71baaf7650f4fa8a1d168c7fb6ee41f0c9" | |
uuid = "c1ae055f-0cd5-4b69-90a6-9a35b1a98df9" | |
version = "0.1.0" | |
[[deps.RecipesBase]] | |
deps = ["SnoopPrecompile"] | |
git-tree-sha1 = "261dddd3b862bd2c940cf6ca4d1c8fe593e457c8" | |
uuid = "3cdcf5f2-1ef4-517c-9805-6587b60abb01" | |
version = "1.3.3" | |
[[deps.Reexport]] | |
git-tree-sha1 = "45e428421666073eab6f2da5c9d310d99bb12f9b" | |
uuid = "189a3867-3050-52da-a836-e630ba90ab69" | |
version = "1.2.2" | |
[[deps.Referenceables]] | |
deps = ["Adapt"] | |
git-tree-sha1 = "e681d3bfa49cd46c3c161505caddf20f0e62aaa9" | |
uuid = "42d2dcc6-99eb-4e98-b66c-637b7d73030e" | |
version = "0.1.2" | |
[[deps.Requires]] | |
deps = ["UUIDs"] | |
git-tree-sha1 = "838a3a4188e2ded87a4f9f184b4b0d78a1e91cb7" | |
uuid = "ae029012-a4dd-5104-9daa-d747884805df" | |
version = "1.3.0" | |
[[deps.Rmath]] | |
deps = ["Random", "Rmath_jll"] | |
git-tree-sha1 = "f65dcb5fa46aee0cf9ed6274ccbd597adc49aa7b" | |
uuid = "79098fc4-a85e-5d69-aa6a-4863f24498fa" | |
version = "0.7.1" | |
[[deps.Rmath_jll]] | |
deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg"] | |
git-tree-sha1 = "6ed52fdd3382cf21947b15e8870ac0ddbff736da" | |
uuid = "f50d1b31-88e8-58de-be2c-1cc44531875f" | |
version = "0.4.0+0" | |
[[deps.Rotations]] | |
deps = ["LinearAlgebra", "Quaternions", "Random", "StaticArrays", "Statistics"] | |
git-tree-sha1 = "9480500060044fd25a1c341da53f34df7443c2f2" | |
uuid = "6038ab10-8711-5258-84ad-4b1120ba62dc" | |
version = "1.3.4" | |
[[deps.SHA]] | |
uuid = "ea8e919c-243c-51af-8825-aaa63cd721ce" | |
version = "0.7.0" | |
[[deps.SentinelArrays]] | |
deps = ["Dates", "Random"] | |
git-tree-sha1 = "c02bd3c9c3fc8463d3591a62a378f90d2d8ab0f3" | |
uuid = "91c51154-3ec4-41a3-a24f-3f23e20d615c" | |
version = "1.3.17" | |
[[deps.Serialization]] | |
uuid = "9e88b42a-f829-5b0c-bbe9-9e923198166b" | |
[[deps.Setfield]] | |
deps = ["ConstructionBase", "Future", "MacroTools", "StaticArraysCore"] | |
git-tree-sha1 = "e2cc6d8c88613c05e1defb55170bf5ff211fbeac" | |
uuid = "efcf1570-3423-57d1-acb7-fd33fddbac46" | |
version = "1.1.1" | |
[[deps.SharedArrays]] | |
deps = ["Distributed", "Mmap", "Random", "Serialization"] | |
uuid = "1a1011a3-84de-559e-8e89-a11a2f7dc383" | |
[[deps.ShowCases]] | |
git-tree-sha1 = "7f534ad62ab2bd48591bdeac81994ea8c445e4a5" | |
uuid = "605ecd9f-84a6-4c9e-81e2-4798472b76a3" | |
version = "0.1.0" | |
[[deps.SimpleBufferStream]] | |
git-tree-sha1 = "874e8867b33a00e784c8a7e4b60afe9e037b74e1" | |
uuid = "777ac1f9-54b0-4bf8-805c-2214025038e7" | |
version = "1.1.0" | |
[[deps.SimpleTraits]] | |
deps = ["InteractiveUtils", "MacroTools"] | |
git-tree-sha1 = "5d7e3f4e11935503d3ecaf7186eac40602e7d231" | |
uuid = "699a6c99-e7fa-54fc-8d76-47d257e15c1d" | |
version = "0.9.4" | |
[[deps.Sixel]] | |
deps = ["Dates", "FileIO", "ImageCore", "IndirectArrays", "OffsetArrays", "REPL", "libsixel_jll"] | |
git-tree-sha1 = "8fb59825be681d451c246a795117f317ecbcaa28" | |
uuid = "45858cf5-a6b0-47a3-bbea-62219f50df47" | |
version = "0.1.2" | |
[[deps.SnoopPrecompile]] | |
deps = ["Preferences"] | |
git-tree-sha1 = "e760a70afdcd461cf01a575947738d359234665c" | |
uuid = "66db9d55-30c0-4569-8b51-7e840670fc0c" | |
version = "1.0.3" | |
[[deps.Snowball]] | |
deps = ["Languages", "Snowball_jll", "WordTokenizers"] | |
git-tree-sha1 = "d38c1ff8a2fca7b1c65a51457dabebef28052399" | |
uuid = "fb8f903a-0164-4e73-9ffe-431110250c3b" | |
version = "0.1.0" | |
[[deps.Snowball_jll]] | |
deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg"] | |
git-tree-sha1 = "6ff3a185a583dca7265cbfcaae1da16aa3b6a962" | |
uuid = "88f46535-a3c0-54f4-998e-4320a1339f51" | |
version = "2.2.0+0" | |
[[deps.Sockets]] | |
uuid = "6462fe0b-24de-5631-8697-dd941f90decc" | |
[[deps.SortingAlgorithms]] | |
deps = ["DataStructures"] | |
git-tree-sha1 = "a4ada03f999bd01b3a25dcaa30b2d929fe537e00" | |
uuid = "a2af1166-a08f-5f64-846c-94a0d3cef48c" | |
version = "1.1.0" | |
[[deps.SparseArrays]] | |
deps = ["Libdl", "LinearAlgebra", "Random", "Serialization", "SuiteSparse_jll"] | |
uuid = "2f01184e-e22b-5df5-ae63-d93ebab69eaf" | |
[[deps.SpecialFunctions]] | |
deps = ["ChainRulesCore", "IrrationalConstants", "LogExpFunctions", "OpenLibm_jll", "OpenSpecFun_jll"] | |
git-tree-sha1 = "d75bda01f8c31ebb72df80a46c88b25d1c79c56d" | |
uuid = "276daf66-3868-5448-9aa4-cd146d93841b" | |
version = "2.1.7" | |
[[deps.SplittablesBase]] | |
deps = ["Setfield", "Test"] | |
git-tree-sha1 = "e08a62abc517eb79667d0a29dc08a3b589516bb5" | |
uuid = "171d559e-b47b-412a-8079-5efa626c420e" | |
version = "0.1.15" | |
[[deps.StackViews]] | |
deps = ["OffsetArrays"] | |
git-tree-sha1 = "46e589465204cd0c08b4bd97385e4fa79a0c770c" | |
uuid = "cae243ae-269e-4f55-b966-ac2d0dc13c15" | |
version = "0.1.1" | |
[[deps.StaticArrays]] | |
deps = ["LinearAlgebra", "Random", "StaticArraysCore", "Statistics"] | |
git-tree-sha1 = "6954a456979f23d05085727adb17c4551c19ecd1" | |
uuid = "90137ffa-7385-5640-81b9-e52037218182" | |
version = "1.5.12" | |
[[deps.StaticArraysCore]] | |
git-tree-sha1 = "6b7ba252635a5eff6a0b0664a41ee140a1c9e72a" | |
uuid = "1e83bf80-4336-4d27-bf5d-d5a4f845583c" | |
version = "1.4.0" | |
[[deps.Statistics]] | |
deps = ["LinearAlgebra", "SparseArrays"] | |
uuid = "10745b16-79ce-11e8-11f9-7d13ad32a3b2" | |
version = "1.9.0" | |
[[deps.StatsAPI]] | |
deps = ["LinearAlgebra"] | |
git-tree-sha1 = "f9af7f195fb13589dd2e2d57fdb401717d2eb1f6" | |
uuid = "82ae8749-77ed-4fe6-ae5f-f523153014b0" | |
version = "1.5.0" | |
[[deps.StatsBase]] | |
deps = ["DataAPI", "DataStructures", "LinearAlgebra", "LogExpFunctions", "Missings", "Printf", "Random", "SortingAlgorithms", "SparseArrays", "Statistics", "StatsAPI"] | |
git-tree-sha1 = "d1bf48bfcc554a3761a133fe3a9bb01488e06916" | |
uuid = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91" | |
version = "0.33.21" | |
[[deps.StatsFuns]] | |
deps = ["ChainRulesCore", "HypergeometricFunctions", "InverseFunctions", "IrrationalConstants", "LogExpFunctions", "Reexport", "Rmath", "SpecialFunctions"] | |
git-tree-sha1 = "ab6083f09b3e617e34a956b43e9d51b824206932" | |
uuid = "4c63d2b9-4356-54db-8cca-17b64c39e42c" | |
version = "1.1.1" | |
[[deps.StrTables]] | |
deps = ["Dates"] | |
git-tree-sha1 = "5998faae8c6308acc25c25896562a1e66a3bb038" | |
uuid = "9700d1a9-a7c8-5760-9816-a99fda30bb8f" | |
version = "1.0.1" | |
[[deps.Strided]] | |
deps = ["LinearAlgebra", "TupleTools"] | |
git-tree-sha1 = "a7a664c91104329c88222aa20264e1a05b6ad138" | |
uuid = "5e0ebb24-38b0-5f93-81fe-25c709ecae67" | |
version = "1.2.3" | |
[[deps.StringEncodings]] | |
deps = ["Libiconv_jll"] | |
git-tree-sha1 = "33c0da881af3248dafefb939a21694b97cfece76" | |
uuid = "69024149-9ee7-55f6-a4c4-859efe599b68" | |
version = "0.3.6" | |
[[deps.StructArrays]] | |
deps = ["Adapt", "DataAPI", "StaticArrays", "Tables"] | |
git-tree-sha1 = "9abba8f8fb8458e9adf07c8a2377a070674a24f1" | |
uuid = "09ab397b-f2b6-538f-b94a-2f83cf4a842a" | |
version = "0.6.8" | |
[[deps.StructTypes]] | |
deps = ["Dates", "UUIDs"] | |
git-tree-sha1 = "ca4bccb03acf9faaf4137a9abc1881ed1841aa70" | |
uuid = "856f2bd8-1eba-4b0a-8007-ebc267875bd4" | |
version = "1.10.0" | |
[[deps.SuiteSparse]] | |
deps = ["Libdl", "LinearAlgebra", "Serialization", "SparseArrays"] | |
uuid = "4607b0f0-06f3-5cda-b6b1-a6196a1729e9" | |
[[deps.SuiteSparse_jll]] | |
deps = ["Artifacts", "Libdl", "Pkg", "libblastrampoline_jll"] | |
uuid = "bea87d4a-7f5b-5778-9afe-8cc45184846c" | |
version = "5.10.1+0" | |
[[deps.TOML]] | |
deps = ["Dates"] | |
uuid = "fa267f1f-6049-4f14-aa54-33bafae1ed76" | |
version = "1.0.3" | |
[[deps.TableTraits]] | |
deps = ["IteratorInterfaceExtensions"] | |
git-tree-sha1 = "c06b2f539df1c6efa794486abfb6ed2022561a39" | |
uuid = "3783bdb8-4a98-5b6b-af9a-565f29a5fe9c" | |
version = "1.0.1" | |
[[deps.Tables]] | |
deps = ["DataAPI", "DataValueInterfaces", "IteratorInterfaceExtensions", "LinearAlgebra", "OrderedCollections", "TableTraits", "Test"] | |
git-tree-sha1 = "c79322d36826aa2f4fd8ecfa96ddb47b174ac78d" | |
uuid = "bd369af6-aec1-5ad0-b16a-f7cc5008161c" | |
version = "1.10.0" | |
[[deps.Tar]] | |
deps = ["ArgTools", "SHA"] | |
uuid = "a4e569a6-e804-4fa4-b0f3-eef7a1d5b13e" | |
version = "1.10.0" | |
[[deps.TensorBoardLogger]] | |
deps = ["CRC32c", "FileIO", "ImageCore", "ProtoBuf", "Requires", "StatsBase"] | |
git-tree-sha1 = "70205612422b8b8f2e1525d0eb0e9a9df4d260f4" | |
uuid = "899adc3e-224a-11e9-021f-63837185c80f" | |
version = "0.1.19" | |
[[deps.TensorCore]] | |
deps = ["LinearAlgebra"] | |
git-tree-sha1 = "1feb45f88d133a655e001435632f019a9a1bcdb6" | |
uuid = "62fd8b95-f654-4bbd-a8a5-9c27f68ccd50" | |
version = "0.1.1" | |
[[deps.Test]] | |
deps = ["InteractiveUtils", "Logging", "Random", "Serialization"] | |
uuid = "8dfed614-e22c-5e08-85e1-65c5234f0b40" | |
[[deps.TextAnalysis]] | |
deps = ["DataStructures", "DelimitedFiles", "JSON", "Languages", "LinearAlgebra", "Printf", "ProgressMeter", "Random", "Serialization", "Snowball", "SparseArrays", "Statistics", "StatsBase", "Tables", "WordTokenizers"] | |
git-tree-sha1 = "bc85e54209c30e69e1925460ec0257a916683f59" | |
uuid = "a2db99b7-8b79-58f8-94bf-bbc811eef33d" | |
version = "0.7.3" | |
[[deps.ThreadsX]] | |
deps = ["ArgCheck", "BangBang", "ConstructionBase", "InitialValues", "MicroCollections", "Referenceables", "Setfield", "SplittablesBase", "Transducers"] | |
git-tree-sha1 = "34e6bcf36b9ed5d56489600cf9f3c16843fa2aa2" | |
uuid = "ac1d9e8a-700a-412c-b207-f0111f4b6c0d" | |
version = "0.1.11" | |
[[deps.TiffImages]] | |
deps = ["ColorTypes", "DataStructures", "DocStringExtensions", "FileIO", "FixedPointNumbers", "IndirectArrays", "Inflate", "Mmap", "OffsetArrays", "PkgVersion", "ProgressMeter", "UUIDs"] | |
git-tree-sha1 = "7e6b0e3e571be0b4dd4d2a9a3a83b65c04351ccc" | |
uuid = "731e570b-9d59-4bfa-96dc-6df516fadf69" | |
version = "0.6.3" | |
[[deps.TimerOutputs]] | |
deps = ["ExprTools", "Printf"] | |
git-tree-sha1 = "f2fd3f288dfc6f507b0c3a2eb3bac009251e548b" | |
uuid = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f" | |
version = "0.5.22" | |
[[deps.TranscodingStreams]] | |
deps = ["Random", "Test"] | |
git-tree-sha1 = "94f38103c984f89cf77c402f2a68dbd870f8165f" | |
uuid = "3bb67fe8-82b1-5028-8e26-92a6c54297fa" | |
version = "0.9.11" | |
[[deps.Transducers]] | |
deps = ["Adapt", "ArgCheck", "BangBang", "Baselet", "CompositionsBase", "DefineSingletons", "Distributed", "InitialValues", "Logging", "Markdown", "MicroCollections", "Requires", "Setfield", "SplittablesBase", "Tables"] | |
git-tree-sha1 = "c42fa452a60f022e9e087823b47e5a5f8adc53d5" | |
uuid = "28d57a85-8fef-5791-bfe6-a80928e7c999" | |
version = "0.4.75" | |
[[deps.TupleTools]] | |
git-tree-sha1 = "3c712976c47707ff893cf6ba4354aa14db1d8938" | |
uuid = "9d95972d-f1c8-5527-a6e0-b4b365fa01f6" | |
version = "1.3.0" | |
[[deps.URIs]] | |
git-tree-sha1 = "ac00576f90d8a259f2c9d823e91d1de3fd44d348" | |
uuid = "5c2747f8-b7ea-4ff2-ba2e-563bfd36b1d4" | |
version = "1.4.1" | |
[[deps.UUIDs]] | |
deps = ["Random", "SHA"] | |
uuid = "cf7118a7-6976-5b1a-9a39-7adc72f591a4" | |
[[deps.UnPack]] | |
git-tree-sha1 = "387c1f73762231e86e0c9c5443ce3b4a0a9a0c2b" | |
uuid = "3a884ed6-31ef-47d7-9d2a-63182c4928ed" | |
version = "1.0.2" | |
[[deps.Unicode]] | |
uuid = "4ec0a83e-493e-50e2-b9ac-8f72acf5a8f5" | |
[[deps.UnicodePlots]] | |
deps = ["ColorSchemes", "ColorTypes", "Contour", "Crayons", "Dates", "FileIO", "FreeType", "LinearAlgebra", "MarchingCubes", "NaNMath", "Printf", "Requires", "SnoopPrecompile", "SparseArrays", "StaticArrays", "StatsBase", "Unitful"] | |
git-tree-sha1 = "64b6f4f8ffd783af63819126cb3091f97c3b1aec" | |
uuid = "b8865327-cd53-5732-bb35-84acbb429228" | |
version = "3.3.4" | |
[[deps.Unitful]] | |
deps = ["ConstructionBase", "Dates", "LinearAlgebra", "Random"] | |
git-tree-sha1 = "d670a70dd3cdbe1c1186f2f17c9a68a7ec24838c" | |
uuid = "1986cc42-f94f-5a68-af5c-568840ba703d" | |
version = "1.12.2" | |
[[deps.ValueHistories]] | |
deps = ["DataStructures", "RecipesBase"] | |
git-tree-sha1 = "9cc583107e0c09b8986ac10660653fa2e37c0f62" | |
uuid = "98cad3c8-aec3-5f06-8e41-884608649ab7" | |
version = "0.5.4" | |
[[deps.VersionParsing]] | |
git-tree-sha1 = "58d6e80b4ee071f5efd07fda82cb9fbe17200868" | |
uuid = "81def892-9a0e-5fdd-b105-ffc91e053289" | |
version = "1.3.0" | |
[[deps.WeakRefStrings]] | |
deps = ["DataAPI", "InlineStrings", "Parsers"] | |
git-tree-sha1 = "b1be2855ed9ed8eac54e5caff2afcdb442d52c23" | |
uuid = "ea10d353-3f73-51f8-a26c-33c1cb351aa5" | |
version = "1.4.2" | |
[[deps.WoodburyMatrices]] | |
deps = ["LinearAlgebra", "SparseArrays"] | |
git-tree-sha1 = "de67fa59e33ad156a590055375a30b23c40299d3" | |
uuid = "efce3f68-66dc-5838-9240-27a6d6f5f9b6" | |
version = "0.5.5" | |
[[deps.WordTokenizers]] | |
deps = ["DataDeps", "HTML_Entities", "StrTables", "Unicode"] | |
git-tree-sha1 = "01dd4068c638da2431269f49a5964bf42ff6c9d2" | |
uuid = "796a5d58-b03d-544a-977e-18100b691f6e" | |
version = "0.5.6" | |
[[deps.WorkerUtilities]] | |
git-tree-sha1 = "cd1659ba0d57b71a464a29e64dbc67cfe83d54e7" | |
uuid = "76eceee3-57b5-4d4a-8e66-0e911cebbf60" | |
version = "1.6.1" | |
[[deps.XTermColors]] | |
deps = ["Crayons", "ImageBase", "OffsetArrays"] | |
git-tree-sha1 = "bc27b7622a51f570c57b80bd839d1c0d43605b38" | |
uuid = "c8c2cc18-de81-4e68-b407-38a3a0c0491f" | |
version = "0.2.1" | |
[[deps.ZipFile]] | |
deps = ["Libdl", "Printf", "Zlib_jll"] | |
git-tree-sha1 = "f492b7fe1698e623024e873244f10d89c95c340a" | |
uuid = "a5390f91-8eb1-5f08-bee0-b1d1ffed6cea" | |
version = "0.10.1" | |
[[deps.Zlib_jll]] | |
deps = ["Libdl"] | |
uuid = "83775a58-1f1d-513f-b197-d71354ab007a" | |
version = "1.2.13+0" | |
[[deps.Zygote]] | |
deps = ["AbstractFFTs", "ChainRules", "ChainRulesCore", "DiffRules", "Distributed", "FillArrays", "ForwardDiff", "GPUArrays", "GPUArraysCore", "IRTools", "InteractiveUtils", "LinearAlgebra", "LogExpFunctions", "MacroTools", "NaNMath", "Random", "Requires", "SparseArrays", "SpecialFunctions", "Statistics", "ZygoteRules"] | |
git-tree-sha1 = "8ac61a92a33b3fd2a4cbf92951817831e313a004" | |
uuid = "e88e6eb3-aa80-5325-afca-941959d7151f" | |
version = "0.6.44" | |
[[deps.ZygoteRules]] | |
deps = ["MacroTools"] | |
git-tree-sha1 = "8c1a8e4dfacb1fd631745552c8db35d0deb09ea0" | |
uuid = "700de1a5-db45-46bc-99cf-38207098b444" | |
version = "0.2.2" | |
[[deps.libblastrampoline_jll]] | |
deps = ["Artifacts", "Libdl"] | |
uuid = "8e850b90-86db-534c-a0d3-1478176c7d93" | |
version = "5.2.0+0" | |
[[deps.libpng_jll]] | |
deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg", "Zlib_jll"] | |
git-tree-sha1 = "94d180a6d2b5e55e447e2d27a29ed04fe79eb30c" | |
uuid = "b53b4c65-9356-5827-b1ea-8c7a1a84506f" | |
version = "1.6.38+0" | |
[[deps.libsixel_jll]] | |
deps = ["Artifacts", "JLLWrappers", "JpegTurbo_jll", "Libdl", "Pkg", "libpng_jll"] | |
git-tree-sha1 = "d4f63314c8aa1e48cd22aa0c17ed76cd1ae48c3c" | |
uuid = "075b6546-f08a-558a-be8f-8157d0f608a5" | |
version = "1.10.3+0" | |
[[deps.nghttp2_jll]] | |
deps = ["Artifacts", "Libdl"] | |
uuid = "8e850ede-7688-5339-a07c-302acd2aaf8d" | |
version = "1.48.0+0" | |
[[deps.p7zip_jll]] | |
deps = ["Artifacts", "Libdl"] | |
uuid = "3f19e933-33d8-53b3-aaab-bd5110c3b7a0" | |
version = "17.4.0+0" | |
[[deps.protoc_jll]] | |
deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg"] | |
git-tree-sha1 = "f163144b28db53c78ec0ff2b82cfa225f0c7a0f4" | |
uuid = "c7845625-083e-5bbe-8504-b32d602b7110" | |
version = "3.21.12+0" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name = "DisentanglingVAE" | |
uuid = "1739f45c-9d92-4b55-89bf-86046b80dc74" | |
authors = ["Romeo Valentin <romeov@stanford.edu>"] | |
version = "0.1.0" | |
[deps] | |
ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4" | |
FastAI = "5d0beca9-ade8-49ae-ad0b-a3cf890e669f" | |
FastTimm = "6fe279a3-76fd-4f59-807e-d8e68e6f60cf" | |
FastVision = "7bf02486-ff4c-4e73-b158-40c00866b54f" | |
FoldsThreads = "9c68100b-dfe1-47cf-94c8-95104e173443" | |
ImageCore = "a09fc81d-aa75-5fe9-8630-4744c3626534" | |
Metalhead = "dbeba491-748d-5e0e-a39e-b530a07fa0cc" | |
PyCall = "438e738f-606a-5dbb-bf0a-cddfbfd45ab0" | |
PyNNTraining = "013e1a4a-88fe-473f-be94-6c6af62e62ad" | |
ThreadsX = "ac1d9e8a-700a-412c-b207-f0111f4b6c0d" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
using Statistics: mean | |
using FastAI | |
using FastAI.FluxTraining | |
using FastAI.Flux | |
using FastAI: encodedblock, encodedblockfilled, decodedblockfilled | |
import FastAI.Flux.MLUtils: _default_executor | |
import FastAI.MLUtils.Transducers: ThreadedEx | |
using FastVision | |
import FastVision: RGB | |
using FastTimm | |
using Metalhead | |
using PyCall, PyNNTraining | |
# ThreadPoolEx gave me problems, see https://github.com/JuliaML/MLUtils.jl/issues/142 | |
_default_executor() = ThreadedEx() | |
DEVICE = gpu | |
#### Prepare dataset and task ###################################### | |
path = load(datasets()["mnist_tiny"]) | |
data = Datasets.loadfolderdata( | |
path, | |
filterfn = FastVision.isimagefile, | |
loadfn = loadfile, | |
) | |
function EmbeddingTask(block, enc) | |
sample = block | |
encodedsample = x = y = ŷ = encodedblockfilled(enc, block) | |
blocks = (; sample, x, y, ŷ, encodedsample) | |
return BlockTask(blocks, enc) | |
end | |
task = EmbeddingTask(Image{2}(), | |
(ProjectiveTransforms((64, 64)), | |
ImagePreprocessing(means = FastVision.IMAGENET_MEANS, | |
stds = FastVision.IMAGENET_STDS, | |
C = RGB{Float32}, | |
buffered=false, | |
),)) | |
BATCHSIZE=2 | |
dl, dl_val = taskdataloaders(data, task, BATCHSIZE, pctgval=0.1; | |
buffer=true, partial=false); | |
#################################################################### | |
##### Construct model ##### | |
struct AE{E, B, D} | |
encoder::E # this will be a timm model | |
bridge ::B # to bridge the timm model to the latent space | |
decoder::D | |
end | |
Flux.@functor AE | |
# image size is (64, 64) | |
Dhidden_effnet = 1792 | |
Dlatent = 64 | |
# here we load the timm model and remove the classification part | |
torch, functorch = pyimport("torch"), pyimport("functorch.experimental") | |
backbone = load(models()["timm/efficientnetv2_rw_s"], pretrained=true); | |
backbone.classifier = torch.nn.Identity() # 1792 dimensions | |
functorch.replace_all_batch_norm_modules_(backbone); | |
bridge = Dense(Dhidden_effnet, Dlatent) |> DEVICE | |
decoder = Chain(Dense(Dlatent, 16*16*32), | |
xs -> reshape(xs, 16, 16, 32, :), | |
Metalhead.basicblock(32, 128), | |
PixelShuffle(2), | |
Metalhead.basicblock(32, 128), | |
PixelShuffle(2), | |
Metalhead.basicblock(32, 3), | |
) |> DEVICE | |
model = AE(backbone, bridge, decoder); | |
########################### | |
#### Use FluxTraining to define how this model is run. ##### | |
struct AETrainingPhase <: FluxTraining.AbstractTrainingPhase end | |
function FluxTraining.on( | |
::FluxTraining.StepBegin, | |
::AETrainingPhase, | |
cb::ToDevice, | |
learner, | |
) | |
learner.step.x = cb.movedatafn(learner.step.x) | |
end | |
function FluxTraining.step!(learner, phase::AETrainingPhase, batch) | |
FluxTraining.runstep(learner, phase, (x = batch,)) do handle, state | |
gs = gradient(learner.params) do | |
intermediate = learner.model.encoder(state.x) | |
state.z = learner.model.bridge(intermediate) | |
state.x̄ = learner.model.decoder(state.z) | |
handle(FluxTraining.LossBegin()) | |
state.loss = learner.lossfn(state.x, state.x̄) | |
handle(FluxTraining.BackwardBegin()) | |
return state.loss | |
end | |
handle(FluxTraining.BackwardEnd()) | |
Flux.Optimise.update!(learner.optimizer, learner.params, gs) | |
end | |
end | |
############################################################ | |
#### Try to run the training. ####################### | |
reconstruction_error(x, x̄) = mean(sum(@. ((x̄ - x)^2); dims=1)) | |
# This seems to segfault randomly after a while. | |
learner = Learner(model, reconstruction_error, callbacks=[ToPyTorch()]) | |
fitonecycle!(learner, 10, 1e-4; phases=(AETrainingPhase() => dl,)) | |
##################################################### |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment