Skip to content

Releases: pfnet/pytorch-pfn-extras

v0.7.7

24 Jul 10:41
cb41996
Compare
Choose a tag to compare

What's Changed

New Contributors

Full Changelog: v0.7.6...v0.7.7

v0.7.6

19 Jan 05:42
2ca6b8f
Compare
Choose a tag to compare

Enhancements and bug-fixes

  • ShardedSnapshot by @linshokaku in #799
    • e.g. FSDP model support for state_dict/load_state_dict
  • Accumulate Extensions by @linshokaku in #761
    • Provides an aggregate extension of log values in average/min/max/std, etc.
    • Provides aggregation of values between processes in a distributed environment
  • Running pytest in an MPI environment by @linshokaku in #763
  • Append mode ChromeTracer by @linshokaku in #801
  • Change output file destination in pytest to a temporary directory by @linshokaku in #802
  • Fix Logic.consume_options() by @linshokaku in #769
  • fix test by @linshokaku

See the list of merged pull-requests for the details.

v0.7.5

08 Dec 05:34
347d1c9
Compare
Choose a tag to compare

Enhancements and bug-fixes

This release includes the following enhancements and bug fixes:

See the list of merged pull-requests for the details.

v0.7.4

02 Nov 05:19
dd8ed07
Compare
Choose a tag to compare

What's Changed

  • Support for ubuntu 18.04 has been discontinued due to EOL by @linshokaku in #765
  • Fixed an issue where tf32 was implicitly used for tests related to onnx by @take-cheeze in #768
  • pfto float8 support by @niboshi in #781
  • Fixed possible file corruption when running DistributedSnapshot.on_error() by @linshokaku in #780

New Contributors

Full Changelog: v0.7.3...v0.7.4

v0.7.3

05 Oct 08:33
266e8bd
Compare
Choose a tag to compare

Enhancements and bug-fixes

This release includes the following enhancements and bug fixes:

  • Add distributed option to the create_evaluator function to support creating DistributedEvaluator (#675, #733)

See the list of merged pull-requests for the details.

v0.7.2

31 Aug 08:06
55db658
Compare
Choose a tag to compare

Enhancements and bug-fixes

This release includes the following enhancements and bug fixes:

  • Add wait_for_first_optimizer_step option to LRScheduler extension that prevents stepping LR schedulers before stepping optimizers. #737
  • Fix IntervalTrigger trigger to correctly handle an epoch number given in float. #738
  • Fix Snapshot extension to only trigger the cleanup process in the saver rank. #732
  • Improve ppe.cuda.stream to propagate PyTorch's device ID to CuPy. #745
  • Add Quick Start documentation. #743
  • Improved API Reference. #731

See the list of merged pull-requests for the details.

v0.7.1

24 Jul 05:49
bc94bc1
Compare
Choose a tag to compare

Enhancements and bug-fixes

This release includes the following enhancements and bug fixes:

  • Fix CUDA profiling with ppe.profiler to respect the current device #726

See the list of merged pull-requests for the details.

v0.7.0

20 Jul 05:28
9f68c03
Compare
Choose a tag to compare

Enhancements and bug-fixes

This release includes the following enhancements and bug fixes:

  • ppe.compile API to produce computational graphs with joint backward and optimizer steps #714
  • Snapshot support for objects with state_dict passed in the options kwarg to Trainer and Manager #702
  • Evaluator with float16 support #698
  • Multiple fixes to pfto #669 #684 #687 #715
  • Fixed a problem in which autoload of snapshot could not be read when a Writer is specified. #692
  • Fixed a problem that caused model train/eval mode switching after Evaluator execution to not work correctly. #697

No compatibility changes

  • Support for PyTorch 1.9 and Python 3.7 has been removed from the code base #706

See the list of merged pull-requests for the details.

v0.6.7

11 May 07:41
bcb5f37
Compare
Choose a tag to compare

Enhancements and bug-fixes

This release includes the following enhancements and bug fixes:

  • Use state_dict to decide name of initializers #670
  • Fix autocast options consumption with trainer #668
  • Various PFTO fixes

See the list of merged pull-requests for the details.

v0.6.6

06 Apr 09:21
8a10d18
Compare
Choose a tag to compare

Enhancements and bug-fixes

This release includes the following enhancements and bug fixes:

  • Support PyTorch 2.0 #645
  • Add a Logic class for clousure mode optimization #657
  • Enable new PyTorch autocast options #649

See the list of merged pull-requests for the details.