Skip to content
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
merged 54 commits into from
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 Aug 28, 2020
518ba8c
inception v3
rishabhvarshney14 Nov 22, 2020
b829363
corrected model name
rishabhvarshney14 Nov 23, 2020
2ca5282
added SqueezeNet and reduce code repetition
rishabhvarshney14 Nov 26, 2020
e5ef7a8
added DenseNet
rishabhvarshney14 Nov 26, 2020
749f510
some minor changes
rishabhvarshney14 Nov 26, 2020
64376e3
Update to latest Flux
Jan 12, 2021
be80077
Add initial training routine
Jan 14, 2021
5c3cafb
Add updated training script for distributed
Jan 24, 2021
5cc32c4
Updated scripts and README for training
darsnack Jan 24, 2021
0041b88
Updated to latest Flux constructs
Jan 24, 2021
5c2b9f2
Fix Flux compat
darsnack Jan 26, 2021
ffa1544
Updated for released Flux
darsnack Jan 26, 2021
4bd498f
Use Flux#master for now
darsnack Jan 27, 2021
0960dd1
Transfer weights instead
darsnack Feb 15, 2021
7a5c384
Fix resnet to exclude bias from conv layers
darsnack Feb 22, 2021
7d21ed5
Removed BN from googlenet
darsnack Feb 22, 2021
c8894ce
Fix typos in googlenet
darsnack Feb 22, 2021
fe16624
Fix densenet names
darsnack Feb 23, 2021
e37ac87
Add pretrained weight loading
darsnack Mar 2, 2021
f1d0597
Remove training folder for now
darsnack Mar 2, 2021
71a0013
Add pretrained tests and update BSON
darsnack Mar 8, 2021
0dcf081
Add doc strings
darsnack Mar 8, 2021
250ea74
Adjust CI
darsnack Mar 8, 2021
8ad4878
Add initial docs
darsnack Mar 8, 2021
489b410
Add quickstart
darsnack Mar 8, 2021
c5bf016
Update to latest Flux
darsnack Mar 28, 2021
1a48e55
Stop tracking Manifest
darsnack Mar 28, 2021
f4fc9cf
Stop tracking Manifest
darsnack Mar 28, 2021
576f909
Update quickstart
darsnack Mar 28, 2021
0931dfb
Change default initialization
darsnack Mar 28, 2021
dd9975d
Fix alignment of whitespace
darsnack Mar 28, 2021
ab9d76d
Apply suggestions from code review
darsnack Mar 28, 2021
0b86849
Fix file endings
darsnack Mar 28, 2021
3c9f6dd
Add struct versions of models
darsnack Jun 3, 2021
9bb0b30
Update to latest Artifacts system
darsnack Jun 3, 2021
5875374
Add Artifacts/LazyArtifacts to Project.toml
darsnack Jun 3, 2021
d2fbf0f
Fix missing return
darsnack Jun 3, 2021
8ff0a38
Make DenseNet test grouping better
darsnack Jun 3, 2021
ae54958
Add functors
darsnack Jun 11, 2021
feb2241
Fix imports
darsnack Jun 12, 2021
a4a7ab6
Fix functors use
darsnack Jun 16, 2021
c97f4ec
Make output classes configurable
darsnack Jun 18, 2021
906a004
Use FluxML/MetalheadWeights
darsnack Jun 18, 2021
a87d589
Refactor growth rate in DenseNet
darsnack Jul 27, 2021
0f6de0d
Make GoogLeNet inception block internal helper
darsnack Jul 27, 2021
edee644
Remove extra type parameters from VGG/ResNet
darsnack Jul 27, 2021
36005f2
Adjust formatting
darsnack Jul 27, 2021
e4923b0
Big(!) show and test fixes
darsnack Jul 27, 2021
d275c50
Remove extra import
darsnack Jul 29, 2021
37538af
Add gradient tests
darsnack Jul 29, 2021
a587b3d
Break classifier and backbone into two sub-Chains
darsnack Jul 29, 2021
f9e7be5
Reduce BS
darsnack Jul 29, 2021
07925ea
skip backwards pass tests
DhairyaLGandhi Jul 29, 2021
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 4 additions & 11 deletions .github/workflows/CompatHelper.yml
Original file line number Diff line number Diff line change
@@ -7,20 +7,13 @@ on:
types: [opened, reopened]

jobs:
build:
runs-on: ${{ matrix.os }}
strategy:
matrix:
julia-version: [1.2.0]
julia-arch: [x86]
os: [ubuntu-latest]
CompatHelper:
runs-on: ubuntu-latest
steps:
- uses: julia-actions/setup-julia@latest
with:
version: ${{ matrix.julia-version }}
- name: Pkg.add("CompatHelper")
run: julia -e 'using Pkg; Pkg.add("CompatHelper")'
- name: CompatHelper.main()
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
run: julia -e 'using CompatHelper; CompatHelper.main()'
COMPATHELPER_PRIV: ${{ secrets.DOCUMENTER_KEY }}
run: julia -e 'using CompatHelper; CompatHelper.main()'
34 changes: 34 additions & 0 deletions .github/workflows/Publish.yml
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
4 changes: 4 additions & 0 deletions .gitignore
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
39 changes: 39 additions & 0 deletions Artifacts.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
[densenet121]
Copy link
Member

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?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

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"
Loading