here is the debug-log file
2023-07-18 23:03:50,394 INFO MainThread:3851716 [wandb_setup.py:_flush():76] Current SDK version is 0.15.5
2023-07-18 23:03:50,395 INFO MainThread:3851716 [wandb_setup.py:_flush():76] Configure stats pid to 3851716
2023-07-18 23:03:50,395 INFO MainThread:3851716 [wandb_setup.py:_flush():76] Loading settings from /home/ex20249/.config/wandb/settings
2023-07-18 23:03:50,395 INFO MainThread:3851716 [wandb_setup.py:_flush():76] Loading settings from /home/ex20249/xai/src/train/wandb/settings
2023-07-18 23:03:50,395 INFO MainThread:3851716 [wandb_setup.py:_flush():76] Loading settings from environment variables: {}
2023-07-18 23:03:50,395 INFO MainThread:3851716 [wandb_setup.py:_flush():76] Applying setup settings: {‘_disable_service’: False}
2023-07-18 23:03:50,395 INFO MainThread:3851716 [wandb_setup.py:_flush():76] Inferring run settings from compute environment: {‘program_relpath’: ‘train_model.py’, ‘program’: ‘train_model.py’}
2023-07-18 23:03:50,395 INFO MainThread:3851716 [wandb_init.py:_log_setup():507] Logging user logs to /home/ex20249/xai/src/train/wandb/run-20230718_230350-mwxuu6tx/logs/debug.log
2023-07-18 23:03:50,395 INFO MainThread:3851716 [wandb_init.py:_log_setup():508] Logging internal logs to /home/ex20249/xai/src/train/wandb/run-20230718_230350-mwxuu6tx/logs/debug-internal.log
2023-07-18 23:03:50,396 INFO MainThread:3851716 [wandb_init.py:init():547] calling init triggers
2023-07-18 23:03:50,396 INFO MainThread:3851716 [wandb_init.py:init():555] wandb.init called with sweep_config: {}
config: {‘DATASET’: ‘walmart’, ‘NUM_SERIES’: 100, ‘HISTORY_SIZE’: 30, ‘TARGET_SIZE’: 6, ‘STRIDE’: 1, ‘CONT_FEATURES’: [0, 9], ‘CAT_FEATURES’: [1, 2, 3, 4, 5, 6, 7, 8], ‘BATCH_SIZE’: 512, ‘EPOCHS’: 200, ‘PATIENCE’: 10, ‘MODEL’: ‘lstm’, ‘NUM_LAYERS’: 2, ‘NUM_UNITS’: 32, ‘DROPOUT’: 0, ‘STANDARDIZE’: False}
2023-07-18 23:03:50,396 INFO MainThread:3851716 [wandb_init.py:init():596] starting backend
2023-07-18 23:03:50,396 INFO MainThread:3851716 [wandb_init.py:init():600] setting up manager
2023-07-18 23:03:50,400 INFO MainThread:3851716 [backend.py:_multiprocessing_setup():108] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
2023-07-18 23:03:50,401 INFO MainThread:3851716 [wandb_init.py:init():606] backend started and connected
2023-07-18 23:03:50,403 INFO MainThread:3851716 [wandb_init.py:init():705] updated telemetry
2023-07-18 23:03:50,404 INFO MainThread:3851716 [wandb_init.py:init():738] communicating run to backend with 60.0 second timeout
2023-07-18 23:03:50,837 INFO MainThread:3851716 [wandb_run.py:_on_init():2173] communicating current version
2023-07-18 23:03:50,895 INFO MainThread:3851716 [wandb_run.py:_on_init():2182] got version response
2023-07-18 23:03:50,895 INFO MainThread:3851716 [wandb_init.py:init():789] starting run threads in backend
2023-07-18 23:04:20,922 INFO MainThread:3851716 [wandb_run.py:_console_start():2152] atexit reg
2023-07-18 23:04:20,922 INFO MainThread:3851716 [wandb_run.py:_redirect():2007] redirect: SettingsConsole.WRAP_RAW
2023-07-18 23:04:20,922 INFO MainThread:3851716 [wandb_run.py:_redirect():2072] Wrapping output streams.
2023-07-18 23:04:20,922 INFO MainThread:3851716 [wandb_run.py:_redirect():2097] Redirects installed.
2023-07-18 23:04:20,922 INFO MainThread:3851716 [wandb_init.py:init():830] run started, returning control to user process
2023-07-18 23:04:21,665 INFO MainThread:3851716 [wandb_run.py:_config_callback():1281] config_cb None None {‘FEATURES’: [‘Weekly_Sales’, ‘Temperature’, ‘Store’, ‘Dept’, ‘Year’, ‘month’, ‘weekofmonth’, ‘day’, ‘Type’, ‘Size’], ‘CONT_FEATURES’: [0, 9], ‘CAT_FEATURES’: [1, 2, 3, 4, 5, 6, 7, 8]}
By the way, the code could be used in colab.