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

Jump to bottom

Core structure updates: image and text featurizer #172

Draft
wants to merge 17 commits into
base: master
Choose a base branch
from
Draft

Core structure updates: image and text featurizer #172

Changes from 1 commit
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
517f035
new base and sample dataset
zzachw Jun 14, 2023
ba0898b
task template
zzachw Jun 14, 2023
ba27b77
add covid19_cxr dataset and task
zzachw Jun 14, 2023
831c495
add resnet model
zzachw Jun 14, 2023
2e2550d
add medical transcriptions dataset and task
zzachw Jun 14, 2023
c952c59
add connection to huggingface
zzachw Jun 14, 2023
8f13af1
add resnet to models.init
zzachw Jun 17, 2023
f29022e
add huggingfaceautomodel to models.init
zzachw Jun 17, 2023
f438754
nan bug fixed for MedicalTranscriptionsDataset
zzachw Jun 17, 2023
8e88da3
add torchvision_classif.py
wuzijian1997 Jun 22, 2023
52f9d15
Merge pull request #175 from wuzijian1997/zijian_image_classification
zzachw Jun 23, 2023
8055fef
update torchvision model
zzachw Jun 23, 2023
5b1cb90
tmp commit
zzachw Jul 12, 2023
47ee96f
wrap up this pr
zzachw Jul 15, 2023
6f5cf53
update init
zzachw Jul 15, 2023
19e5abd
wrap up the pr: import bug fix
zzachw Jul 16, 2023
ec3c1aa
wrap up the pr: bug fix
zzachw Jul 16, 2023
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
Prev Previous commit
wrap up the pr: bug fix
  • Loading branch information
@zzachw
zzachw committed Jul 16, 2023
commit ec3c1aa4b709cc166ca1ad074ef08cfb3ec78693
2 changes: 1 addition & 1 deletion pyhealth/models/torchvision_model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ def __init__(
hidden_dim = final_layer.in_features
self.label_tokenizer = self.get_label_tokenizer()
output_size = self.get_output_size(self.label_tokenizer)
setattr(self.model, final_layer_name, nn.Linear(hidden_dim, output_size))
setattr(self.model, final_layer_name.split(".")[0], nn.Linear(hidden_dim, output_size))

def forward(self, **kwargs) -> Dict[str, torch.Tensor]:
"""Forward propagation."""
Expand Down