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.

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

Transfer learning #63

Merged
merged 18 commits into from
Jan 16, 2024
Merged

Transfer learning #63

merged 18 commits into from
Jan 16, 2024

Conversation

ArminSchenk
Copy link
Collaborator

No description provided.

ArminSchenk and others added 18 commits October 30, 2023 10:19
- Rework plot function of CNNs
- Fix bug that transformed integer data into torch tensors with a wrong data type
- Added Note when using continue_training on a different device than the original training
- Fix bug that rearranged the levels of a factor when the levels were numbers (e.g. mnist labels)
- Always move the model to cpu after training instead of leaving it on the gpu
…an be generated (important for transfer learning)
…e is specified.

Also fix a bug that prevented dnn() to be used without providing data.
…ly added if output=NULL (happens when no response is specified in dnn() or cnn())
…n output_shape=NULL (no response specified in dnn() or cnn())
@ArminSchenk ArminSchenk merged commit 801a01a into citoverse:main Jan 16, 2024
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant