Skip to content

Commit

Permalink
up
Browse files Browse the repository at this point in the history
  • Loading branch information
jeremiedb committed Feb 27, 2024
1 parent c35dba5 commit 663e0e3
Show file tree
Hide file tree
Showing 3 changed files with 1 addition and 14 deletions.
4 changes: 0 additions & 4 deletions .github/workflows/TagBot.yml
Original file line number Diff line number Diff line change
Expand Up @@ -25,10 +25,6 @@ jobs:
if: github.event_name == 'workflow_dispatch' || github.actor == 'JuliaTagBot'
runs-on: ubuntu-latest
steps:
- uses: webfactory/[email protected]
with:
ssh-private-key: |
${{ secrets.NEUROTREEMODELS_KEY }}
- uses: JuliaRegistries/TagBot@v1
with:
token: ${{ secrets.GITHUB_TOKEN }}
Expand Down
5 changes: 1 addition & 4 deletions .github/workflows/docs.yml
Original file line number Diff line number Diff line change
Expand Up @@ -32,10 +32,6 @@ jobs:
with:
version: '1.10'
- uses: julia-actions/cache@v1
- uses: webfactory/[email protected]
with:
ssh-private-key: |
${{ secrets.NEUROTREEMODELS_KEY }}
- uses: julia-actions/julia-buildpkg@v1
with:
git_cli: true # = JULIA_PKG_USE_CLI_GIT. Options: true | false (default)
Expand All @@ -44,3 +40,4 @@ jobs:
- uses: julia-actions/julia-docdeploy@v1
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
DOCUMENTER_KEY: ${{ secrets.NEUROTREEMODELS_KEY }}
6 changes: 0 additions & 6 deletions src/model.jl
Original file line number Diff line number Diff line change
Expand Up @@ -403,12 +403,6 @@ function get_model_chain(L; config, nfeats)
init_scale=config.init_scale)
)

chain = Chain(
BatchNorm(nfeats),
tree=NeuroTree(hidden_size => hidden_size; depth, ntrees, actA, init_scale),
tree=NeuroTree(hidden_size => hidden_size; depth, ntrees, actA, init_scale),
)

end

return chain
Expand Down

0 comments on commit 663e0e3

Please sign in to comment.