skorch

skorch copied to clipboard

HfHubWriter: Save checkpoints on Hugging Face Hub

5

~This is WIP, more documentation and tests are still missing. This draft should serve to discuss the design.~ _Update_ ## Description This PR adds the possibility to create model checkpoints...

BenjaminBossan

Supersedes #877 ## Description This bug could occur if a user has set a parameter with a CUDA dependency and then tries to load the net without CUDA. Now, this...

BenjaminBossan

Solves #848 ## Description For `Checkpoint`, as of now, `dirname` can only be a string. With this update, it can also be a callable with no arguments that returns a...

BenjaminBossan

Checkpoint callbacks with n_jobs > 1

6

I would to use gridsearchcv in parallel mode. However, I think that the checkpoints used during the trainings in the different processes could override each other between them since they...

andreapi87

ValueError: Classification metrics can't handle a mix of unknown and binary targets

4

Hi, I am using the following callback to get accuracy at the end of each epoch: ``` ('valid_acc', EpochScoring( 'accuracy', name='valid_acc', lower_is_better=False,)), ``` However, at the time of evaluation on...

otoofim

readthedocs search optimization

1

readthedocs analytics says that we have several search results that yield little or no useful results. Let's improvethose: - gpu (only 2 results): make sure that explanation of `device` parameter...

ottonemo

Adding DGL and DGL-LifeSci support

4

Hi, I have been currently working with Skorch for standard NN frameworks on tensors and recently started experimenting with some graph neural networks. In particular dgl and dgl-LifeSci. I have...

BernardoHernandezAdame

Add notebook that shows how to use skorch with optuna

3

Fixes #718 [This notebook](https://github.com/skorch-dev/skorch/blob/feature/optuna-example-718/notebooks/optuna-example.ipynb) shows how to use skorch in conjunction with optuna, which can be much more efficient than `GridSearchCV` or `RandomizedSearchCV`. I don't have a deep knowledge of...

BenjaminBossan

FixMatch (semi-supervised learning) usecase

8

Hi Thanks for the great work and package. I have recently started using skorch :) I would like to implement FixMatch in skorch: https://arxiv.org/abs/2001.07685 so the main idea is that...

aliechoes

Can History be more powerful?

5

The current History class has some limitations: (ver 0.10.0) 1. Currently the history is saved as JSON, as a result, those recorded values are limited to simple numbers and strings....

TheAutumnOfRice