Load File Question

[skyWalker1997]

Hello！I have already read your paper and tried to run your code. But I got a wired problem here. When I tried to run this code on my CPU server(CentOS 7,8G memory),it always killed by system because of out of memory.And it would be kill in sampling stage.（Ubuntu 18 can`t run successfully either,killed when it sampling）. But when I run this code on my macbook,it will not be killed by macos.(Just little slow when train one epoch，about 1 hour) I tried to turn up the virtual memory in CentOS (etc. 10g),it was killed when it smapling 1600 timeslots(when virtual memory is zero,it just can make 900). So how could I fixed this problem?Or must I use GPU to run your model?

BTW, have you ever put the compare experiment code in your project.I guess there is a single "if model_name[2:] == "stdn"" in your code.Maybe here was a "if model_name[2:] == "lstn""before LOL.(If you would offer the compare experiment code I will really appreciate it)

Thx a lot!