Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merge best-exp into main #239

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Merge best-exp into main #239

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
30 changes: 15 additions & 15 deletions dvc.lock
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ stages:
random_seed: 42
train:
valid_pct: 0.1
arch: shufflenet_v2_x2_0
arch: resnet34
img_size: 256
batch_size: 8
fine_tune_args:
Expand All @@ -57,16 +57,16 @@ stages:
outs:
- path: models/model.pkl
hash: md5
md5: 63cb30df484bfa1d6ea2cdebac74876c
size: 201725
md5: 3f9fb1dbc04046ddb6f6a96ac967c606
size: 207334766
- path: models/model.pth
hash: md5
md5: 07b113fe1ab01de2d8a5453dccacdd3e
size: 165147
md5: df30344c2f629c248855169e5f525fc0
size: 207285126
- path: results/train
hash: md5
md5: a26429ff680d9b01c7e92043821bf41c.dir
size: 955
md5: 8d42ff1cae8857bf1a35f7f3ccee041a.dir
size: 957
nfiles: 5
evaluate:
cmd: python src/evaluate.py
Expand All @@ -78,8 +78,8 @@ stages:
nfiles: 24
- path: models/model.pkl
hash: md5
md5: 63cb30df484bfa1d6ea2cdebac74876c
size: 201725
md5: 3f9fb1dbc04046ddb6f6a96ac967c606
size: 207334766
- path: src/evaluate.py
hash: md5
md5: 84d2fd3b371546730396a763a51527a0
Expand All @@ -93,19 +93,19 @@ stages:
outs:
- path: results/evaluate
hash: md5
md5: 34985c391291e22ac0cfdf6e83ec7268.dir
size: 1257936
md5: ae80d36acde3d28b13817c4953a7a94a.dir
size: 1254205
nfiles: 11
sagemaker:
cmd: cp models/model.pth sagemaker/code/model.pth && cd sagemaker && tar -cpzf
model.tar.gz code/ && cd .. && mv sagemaker/model.tar.gz . && rm sagemaker/code/model.pth
deps:
- path: models/model.pth
hash: md5
md5: 07b113fe1ab01de2d8a5453dccacdd3e
size: 165147
md5: df30344c2f629c248855169e5f525fc0
size: 207285126
outs:
- path: model.tar.gz
hash: md5
md5: c615b812da343d71fcda080597a24525
size: 145778
md5: 4ba1d6aa076c3ca720aeac166606aa66
size: 180319319
2 changes: 1 addition & 1 deletion dvc.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ artifacts:
- cv
- segmentation
- satellite-images
- shufflenet_v2_x2_0
- resnet34
stages:
data_split:
cmd: python src/data_split.py
Expand Down
9 changes: 4 additions & 5 deletions params.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,17 +2,16 @@ base:
random_seed: 42

data_split:
test_regions:
- REGION_1

test_regions:
- REGION_1
train:
valid_pct: 0.1
arch: shufflenet_v2_x2_0
arch: resnet34
img_size: 256
batch_size: 8
fine_tune_args:
epochs: 8
base_lr: 0.01
base_lr: 0.01

evaluate:
n_samples_to_save: 10