Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 11, 2023

• Merge pull request #3 from soda-inria/fix/device_validation

  Show description for a749c90
  fcharras

  authored
  a749c90

  Copy full SHA for a749c90

• Pin python to 3.11 for compatibility with current docker cpu binaries

  fcharras

  committed
  33e3af0

  Copy full SHA for 33e3af0

• Fix a bug that forced compute to device set by SKLEARN_PYTORCH_ENGINE_DEFAULT_DEVICE and always CPU if not set

  fcharras

  committed
  06fb434

  Copy full SHA for 06fb434

Commits on Sep 20, 2023

• Fix has_fp64_support for MPS on Apple Silicon

  ogrisel

  committed
  8d6b69d

  Copy full SHA for 8d6b69d

• Merge pull request #2 from soda-inria/fix/torch_choice_for_large_datasets

  Show description for f39ff1d
  fcharras

  authored
  f39ff1d

  Copy full SHA for f39ff1d

• Circumvent a pytorch limitation that prevent sampling in more than 2**24 samples, reuse sklearn strategy ~ are there important precision errors that should be investigated for very large datasets ??

  fcharras

  committed
  6f408b1

  Copy full SHA for 6f408b1

Commits on Sep 12, 2023

• Minor maintenance

  fcharras

  committed
  e962dea

  Copy full SHA for e962dea

Commits on Sep 1, 2023

• Merge pull request #1 from soda-inria/fea/pytorch_kmeans

  Show description for 00ceed1
  fcharras

  authored
  00ceed1

  Copy full SHA for 00ceed1

• typo fixup

  fcharras

  committed
  520067c

  Copy full SHA for 520067c

• Add README

  fcharras

  committed
  6405e2e

  Copy full SHA for 6405e2e

• Run test suite on cpu

  fcharras

  committed
  869c486

  Copy full SHA for 869c486

Commits on Aug 31, 2023

• add _utils.py

  fcharras

  committed
  0eee1d6

  Copy full SHA for 0eee1d6

• Bit of cleaning - KMeans OK

  fcharras

  committed
  599be66

  Copy full SHA for 599be66

• Introduce 2 new environment variables SKLEARN_PYTORCH_ENGINE_DEFAULT_DEVICE and SKLEARN_PYTORCH_ENGINE_TEST_INPUTS_DEVICE and ensure unit tests pass with any combination of those 2 variables

  fcharras

  committed
  6cc4700

  Copy full SHA for 6cc4700

Commits on Aug 28, 2023

• isort happy

  fcharras

  committed
  c182749

  Copy full SHA for c182749

• also pass sklearn tests now

  fcharras

  committed
  d9dfc14

  Copy full SHA for d9dfc14

Commits on Aug 25, 2023

• Pass custom tests

  fcharras

  committed
  29e61a5

  Copy full SHA for 29e61a5

Commits on Aug 24, 2023

• WIP: implementing plugin for pytorch kmeans

  fcharras

  committed
  e27046f

  Copy full SHA for e27046f

• First commit

  fcharras

  committed
  6bac648

  Copy full SHA for 6bac648