a question about train_pom.py

[qq291078365]

how to solve this problem?please
Traceback (most recent call last):
File "train_pom.py"， line 274, in main(PARAMS)
File "train _pom.py", line 248, in main
results = [ahid, vhid,thid，adr，vd, tdr，factor_lr，lr， x， batch_sz, decay， min_valid_loss.cpu().data.numpy()]
AttributeError : 'float' object has no attribute 'cpu'

[songyh111]

how long will pom run?

[Justin1904]

I believe this is caused by this code not compatible with later pytorch versions. Could you try fixing your torch version?