i get this error: [2m[36m(train_model pid=162019)[0m /opt/conda/lib/python3.10/site-packages/torchvision/models/_utils.py:223: UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=ResNet50_Weights.IMAGENET1K_V1`. You can also use `weights=ResNet50_Weights.DEFAULT` to get the most up-to-date weights.
[2m[36m(train_model pid=162019)[0m   warnings.warn(msg)
2023-08-15 23:54:35,992	ERROR tune_controller.py:911 -- Trial task failed for trial train_model_03e70_00002
Traceback (most recent call last):
  File "/opt/conda/lib/python3.10/site-packages/ray/air/execution/_internal/event_manager.py", line 110, in resolve_future
    result = ray.get(future)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/auto_init_hook.py", line 24, in auto_init_wrapper
    return fn(*args, **kwargs)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/client_mode_hook.py", line 103, in wrapper
    return func(*args, **kwargs)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/worker.py", line 2524, in get
    raise value.as_instanceof_cause()
ray.exceptions.RayTaskError(AttributeError): [36mray::ImplicitFunc.train()[39m (pid=162020, ip=172.17.0.2, actor_id=0222c3e29498797797a3046401000000, repr=train_model)
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/trainable.py", line 375, in train
    raise skipped from exception_cause(skipped)
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/function_trainable.py", line 349, in entrypoint
    return self._trainable_func(
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/function_trainable.py", line 666, in _trainable_func
    output = fn()
  File "/tmp/ipykernel_113847/2584916037.py", line 344, in train_model
AttributeError: type object 'Checkpoint' has no attribute 'from_config'
2023-08-15 23:54:36,011	ERROR tune_controller.py:911 -- Trial task failed for trial train_model_03e70_00001
Traceback (most recent call last):
  File "/opt/conda/lib/python3.10/site-packages/ray/air/execution/_internal/event_manager.py", line 110, in resolve_future
    result = ray.get(future)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/auto_init_hook.py", line 24, in auto_init_wrapper
    return fn(*args, **kwargs)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/client_mode_hook.py", line 103, in wrapper
    return func(*args, **kwargs)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/worker.py", line 2524, in get
    raise value.as_instanceof_cause()
ray.exceptions.RayTaskError(AttributeError): [36mray::ImplicitFunc.train()[39m (pid=162019, ip=172.17.0.2, actor_id=8808b55d16455321cdcc498e01000000, repr=train_model)
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/trainable.py", line 375, in train
    raise skipped from exception_cause(skipped)
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/function_trainable.py", line 349, in entrypoint
    return self._trainable_func(
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/function_trainable.py", line 666, in _trainable_func
    output = fn()
  File "/tmp/ipykernel_113847/2584916037.py", line 344, in train_model
AttributeError: type object 'Checkpoint' has no attribute 'from_config'
2023-08-15 23:54:36,156	ERROR tune_controller.py:911 -- Trial task failed for trial train_model_03e70_00000
Traceback (most recent call last):
  File "/opt/conda/lib/python3.10/site-packages/ray/air/execution/_internal/event_manager.py", line 110, in resolve_future
    result = ray.get(future)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/auto_init_hook.py", line 24, in auto_init_wrapper
    return fn(*args, **kwargs)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/client_mode_hook.py", line 103, in wrapper
    return func(*args, **kwargs)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/worker.py", line 2524, in get
    raise value.as_instanceof_cause()
ray.exceptions.RayTaskError(AttributeError): [36mray::ImplicitFunc.train()[39m (pid=162018, ip=172.17.0.2, actor_id=ecf00ae16e1ebd03a3dddd1c01000000, repr=train_model)
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/trainable.py", line 375, in train
    raise skipped from exception_cause(skipped)
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/function_trainable.py", line 349, in entrypoint
    return self._trainable_func(
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/function_trainable.py", line 666, in _trainable_func
    output = fn()
  File "/tmp/ipykernel_113847/2584916037.py", line 344, in train_model
AttributeError: type object 'Checkpoint' has no attribute 'from_config'
2023-08-15 23:54:36,228	ERROR tune_controller.py:911 -- Trial task failed for trial train_model_03e70_00003
Traceback (most recent call last):
  File "/opt/conda/lib/python3.10/site-packages/ray/air/execution/_internal/event_manager.py", line 110, in resolve_future
    result = ray.get(future)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/auto_init_hook.py", line 24, in auto_init_wrapper
    return fn(*args, **kwargs)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/client_mode_hook.py", line 103, in wrapper
    return func(*args, **kwargs)
  File "/opt/conda/lib/python3.10/site-packages/ray/_private/worker.py", line 2524, in get
    raise value.as_instanceof_cause()
ray.exceptions.RayTaskError(AttributeError): [36mray::ImplicitFunc.train()[39m (pid=162021, ip=172.17.0.2, actor_id=54b32ccbcd894e5ee8ceac0e01000000, repr=train_model)
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/trainable.py", line 375, in train
    raise skipped from exception_cause(skipped)
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/function_trainable.py", line 349, in entrypoint
    return self._trainable_func(
  File "/opt/conda/lib/python3.10/site-packages/ray/tune/trainable/function_trainable.py", line 666, in _trainable_func
    output = fn()
  File "/tmp/ipykernel_113847/2584916037.py", line 344, in train_model
AttributeError: type object 'Checkpoint' has no attribute 'from_config'
2023-08-15 23:54:36,237	ERROR tune.py:1144 -- Trials did not complete: [train_model_03e70_00000, train_model_03e70_00001, train_model_03e70_00002, train_model_03e70_00003]
2023-08-15 23:54:36,238	INFO tune.py:1148 -- Total run time: 31.74 seconds (31.72 seconds for the tuning loop).
2023-08-15 23:54:36,243	WARNING experiment_analysis.py:916 -- Failed to read the results for 4 trials:
- /tmp/ray_results/pbt_test/train_model_03e70_00000_0_lr=0.7779,momentum=0.2383_2023-08-15_23-54-04
- /tmp/ray_results/pbt_test/train_model_03e70_00001_1_lr=0.8245,momentum=0.9658_2023-08-15_23-54-04
- /tmp/ray_results/pbt_test/train_model_03e70_00002_2_lr=0.9726,momentum=0.4540_2023-08-15_23-54-04
- /tmp/ray_results/pbt_test/train_model_03e70_00003_3_lr=0.6094,momentum=0.7758_2023-08-15_23-54-04