rtdl_FTTransformer issue with dataset `openml__Click_prediction_small__190408`

[duncanmcelfresh]

example traceback:

Traceback (most recent call last):
  File "/home/shared/tabzilla/TabSurvey/tabzilla_experiment.py", line 137, in __call__
    result = cross_validation(model, self.dataset, self.time_limit)
  File "/home/shared/tabzilla/TabSurvey/tabzilla_utils.py", line 236, in cross_validation
    loss_history, val_loss_history = curr_model.fit(
  File "/home/shared/tabzilla/TabSurvey/models/rtdl.py", line 376, in fit
    return super().fit(X, y, X_val, y_val)
  File "/home/shared/tabzilla/TabSurvey/models/basemodel_torch.py", line 154, in fit
    self.load_model(filename_extension="best", directory=self.tmp_name)
  File "/home/shared/tabzilla/TabSurvey/models/basemodel_torch.py", line 219, in load_model
    state_dict = torch.load(filename)
  File "/opt/conda/envs/torch/lib/python3.10/site-packages/torch/serialization.py", line 699, in load
    with _open_file_like(f, 'rb') as opened_file:
  File "/opt/conda/envs/torch/lib/python3.10/site-packages/torch/serialization.py", line 231, in _open_file_like
    return _open_file(name_or_buffer, mode)
  File "/opt/conda/envs/torch/lib/python3.10/site-packages/torch/serialization.py", line 212, in __init__
    super(_open_file, self).__init__(open(name, mode))
FileNotFoundError: [Errno 2] No such file or directory: 'output/rtdl_FTTransformer/openml__Click_prediction_small__190408/tmp_EUJY27V9PMG3/m_best.pt'