fix: more progress

Author: avik-pal

Diff for: examples/DDIM/Project.toml

@@ -19,7 +19,7 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Reactant = "3c362404-f566-11ee-1572-e11a4b42c853"
 StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
-TensorBoardLogger = "899adc3e-224a-11e9-021f-63837185c80f"
+Wandb = "ad70616a-06c9-5745-b1f1-6a5f42545108"
 
 [compat]
 Comonicon = "1"
@@ -41,4 +41,3 @@ Random = "1.10"
 Reactant = "0.2.21"
 StableRNGs = "1.0.2"
 Statistics = "1.10"
-TensorBoardLogger = "0.1"

Diff for: examples/DDIM/main.jl

@@ -8,8 +8,7 @@
 
 using ConcreteStructs, Comonicon, DataAugmentation, DataDeps, Enzyme, FileIO, ImageCore,
       ImageShow, JLD2, Lux, MLUtils, Optimisers, ParameterSchedulers, ProgressTables,
-      Printf, Random, Reactant, StableRNGs, Statistics
-using TensorBoardLogger: TBLogger, log_value, log_images
+      Printf, Random, Reactant, StableRNGs, Statistics, Wandb
 
 # ## Model Definition