Skip to content

Pytorch support - #276

Draft
cpozogonzalez wants to merge 121 commits into
mainfrom
pytorch_support
Draft

Pytorch support#276
cpozogonzalez wants to merge 121 commits into
mainfrom
pytorch_support

Conversation

@cpozogonzalez

Copy link
Copy Markdown
Collaborator

No description provided.

Comment on lines +19 to +23
tensorflow-version: ['latest', '2.16.*', 'none']
torch-version: ['latest', '2.2.*', 'none']
exclude:
- tensorflow-version: 'none'
torch-version: 'none'

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

why you need none here? We always install both tensorflow/keras and torch flavors.

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Why we are loading the iris dataset and plot it in the pytest?

@tjarkmiener tjarkmiener left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

questions inline

…nd move all the warnings to the debug options
Comment thread ctlearn/core/pytorch/model.py Outdated
Comment on lines +127 to +134
in_channels = input_shape[0]
in_channels = input_shape[-1]

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

I don't understand why this is changed? pytorch input shape is channel first, i.e. (channel, height, width), right? Your data loader should supply the input data in this shape rather than changing the model

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

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

I alredy fix it was a mistake

…, with 2 channels or 1 depend of the input (peak time or not)
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.

2 participants