-
-
Notifications
You must be signed in to change notification settings - Fork 67
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
Refactor repo with more complete models and documentation #70
Merged
DhairyaLGandhi
merged 54 commits into
FluxML:master
from
darsnack:darsnack/vision-refactor
Jul 29, 2021
Merged
Changes from 44 commits
Commits
Show all changes
54 commits
Select commit
Hold shift + click to select a range
602ccec
Added vision models
darsnack 518ba8c
inception v3
rishabhvarshney14 b829363
corrected model name
rishabhvarshney14 2ca5282
added SqueezeNet and reduce code repetition
rishabhvarshney14 e5ef7a8
added DenseNet
rishabhvarshney14 749f510
some minor changes
rishabhvarshney14 64376e3
Update to latest Flux
be80077
Add initial training routine
5c3cafb
Add updated training script for distributed
5cc32c4
Updated scripts and README for training
darsnack 0041b88
Updated to latest Flux constructs
5c2b9f2
Fix Flux compat
darsnack ffa1544
Updated for released Flux
darsnack 4bd498f
Use Flux#master for now
darsnack 0960dd1
Transfer weights instead
darsnack 7a5c384
Fix resnet to exclude bias from conv layers
darsnack 7d21ed5
Removed BN from googlenet
darsnack c8894ce
Fix typos in googlenet
darsnack fe16624
Fix densenet names
darsnack e37ac87
Add pretrained weight loading
darsnack f1d0597
Remove training folder for now
darsnack 71a0013
Add pretrained tests and update BSON
darsnack 0dcf081
Add doc strings
darsnack 250ea74
Adjust CI
darsnack 8ad4878
Add initial docs
darsnack 489b410
Add quickstart
darsnack c5bf016
Update to latest Flux
darsnack 1a48e55
Stop tracking Manifest
darsnack f4fc9cf
Stop tracking Manifest
darsnack 576f909
Update quickstart
darsnack 0931dfb
Change default initialization
darsnack dd9975d
Fix alignment of whitespace
darsnack ab9d76d
Apply suggestions from code review
darsnack 0b86849
Fix file endings
darsnack 3c9f6dd
Add struct versions of models
darsnack 9bb0b30
Update to latest Artifacts system
darsnack 5875374
Add Artifacts/LazyArtifacts to Project.toml
darsnack d2fbf0f
Fix missing return
darsnack 8ff0a38
Make DenseNet test grouping better
darsnack ae54958
Add functors
darsnack feb2241
Fix imports
darsnack a4a7ab6
Fix functors use
darsnack c97f4ec
Make output classes configurable
darsnack 906a004
Use FluxML/MetalheadWeights
darsnack a87d589
Refactor growth rate in DenseNet
darsnack 0f6de0d
Make GoogLeNet inception block internal helper
darsnack edee644
Remove extra type parameters from VGG/ResNet
darsnack 36005f2
Adjust formatting
darsnack e4923b0
Big(!) show and test fixes
darsnack d275c50
Remove extra import
darsnack 37538af
Add gradient tests
darsnack a587b3d
Break classifier and backbone into two sub-Chains
darsnack f9e7be5
Reduce BS
darsnack 07925ea
skip backwards pass tests
DhairyaLGandhi File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
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 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
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,34 @@ | ||
# Adapted from https://github.com/MichaelHatherly/Publish.jl | ||
name: Publish | ||
|
||
on: | ||
push: | ||
branches: ['main'] | ||
|
||
jobs: | ||
publish: | ||
name: "Publish Documentation (dev)" | ||
runs-on: ubuntu-latest | ||
steps: | ||
- uses: actions/checkout@v2 | ||
- uses: actions/checkout@v2 | ||
with: | ||
ref: gh-pages | ||
path: gh-pages | ||
- uses: julia-actions/setup-julia@latest | ||
with: | ||
version: '1.5' | ||
- name: Install dependencies | ||
run: | | ||
julia --color=yes --project=. -e 'using Pkg; Pkg.instantiate()' | ||
- name: Build and deploy | ||
run: | | ||
cd gh-pages | ||
julia --color=yes --project=../docs -e "using Pkg; Pkg.instantiate()" | ||
julia --color=yes --project=../docs -e 'using Pkg; Pkg.develop(path="..")' | ||
julia --color=yes --project=../docs ../docs/make.jl | ||
git config user.name github-actions | ||
git config user.email github-actions@github.com | ||
git add . | ||
git commit -m "(dev) Built by Publish.jl." | ||
git push |
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
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -2,3 +2,7 @@ | |
*.jl.*.cov | ||
*.jl.mem | ||
*.bson | ||
|
||
# manifests | ||
docs/Manifest.toml | ||
Manifest.toml |
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
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,39 @@ | ||
[densenet121] | ||
git-tree-sha1 = "ffc7f7ed1e7f67baca4b76f6c100e0d5042ff063" | ||
lazy = true | ||
|
||
[[densenet121.download]] | ||
sha256 = "3fd10f0be70cf072fa7f1358f1fbbe01138440dbcaec1b7c8e007084382c1557" | ||
url = "https://github.com/FluxML/MetalheadWeights/releases/download/v0.1.1/densenet121-0.1.1.tar.gz" | ||
|
||
[googlenet] | ||
git-tree-sha1 = "56cc81845fcca30508fe81da18c7ba0d96d72cdd" | ||
lazy = true | ||
|
||
[[googlenet.download]] | ||
sha256 = "8ab8d60cc26e81451473badc9dc749b5ffc170a11bc00fb4b203da34fbfdc996" | ||
url = "https://github.com/FluxML/MetalheadWeights/releases/download/v0.1.1/googlenet-0.1.1.tar.gz" | ||
|
||
[resnet50] | ||
git-tree-sha1 = "ea3effeaf1ea3969ed5c609f5db5cd0e456ce799" | ||
lazy = true | ||
|
||
[[resnet50.download]] | ||
sha256 = "17760ae50e3d59ed7d74c3dfcdb9f0eeaccec1e2ccd095663955c9fed4f318a8" | ||
url = "https://github.com/FluxML/MetalheadWeights/releases/download/v0.1.1/resnet50-0.1.1.tar.gz" | ||
|
||
[squeezenet] | ||
git-tree-sha1 = "e0e53eb402efe4693417db8cbcc31519e74c8c74" | ||
lazy = true | ||
|
||
[[squeezenet.download]] | ||
sha256 = "a3e60f2731296cdf0f32b79badd227eb8dad88a9bee8c828dbe60382869c50f0" | ||
url = "https://github.com/FluxML/MetalheadWeights/releases/download/v0.1.1/squeezenet-0.1.1.tar.gz" | ||
|
||
[vgg19] | ||
git-tree-sha1 = "072056ec63bf7308cf89885e91852666e191e80a" | ||
lazy = true | ||
|
||
[[vgg19.download]] | ||
sha256 = "0fa000609965604b9d249e84190c30d067d443d73e6c8e340ef09bd013d0bc90" | ||
url = "https://github.com/FluxML/MetalheadWeights/releases/download/v0.1.1/vgg19-0.1.1.tar.gz" |
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Which script was used to generate these?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
https://github.com/FluxML/MetalheadWeights contains the scripts