carefree-learn 0.1.10

Release Notes

carefree-learn 0.1.10 improved overall performances and deepspeed accessibility.

Versioning

carefree-learn now supports checking its version via __version__:

import cflearn

cflearn.__version__  # '0.1.10'

Distributed Training

carefree-learn now provides out-of-the-box API for distributed training with deepspeed:

import cflearn
import numpy as np

x = np.random.random([1000000, 10])
y = np.random.random([1000000, 1])
m = cflearn.deepspeed(x, y, cuda="0,1,2,3").m

⚠️⚠️⚠️ However it is not recommended to use this API unless you have to (e.g. when training some very large models). ⚠️⚠️⚠️

Misc

• Supported use_final_bn in FCNNHead (#75).
• Ensured that models are always in eval mode in inference (#67).
• Supported specifying resource_config of Parallel in Experiment (#68).
• Implemented profile_forward for Pipeline.

---

• Fixed other bugs.
• Accelerated DNDF with Function (#69).
• Patience of TrainMonitor will now depend on dataset size (#43).
• Checkpoints will be logged earlier now when using warmup (#74).