The notebook happens two error.
Oen is the 7th cell
ds_train, ds_valid = ds_train.split(random_state=random.seed(seed_num))
random.seed() needs param name in my environment.
ds_train, ds_valid = ds_train.split(random_state=random.seed(a=seed_num))
Another is the 33rd cell. Commented weights_L1, weights_L2, lines lack cpu(). calling