Thanks for your code! But I got a confusing problem: "TypeError: join() argument must be str or bytes, not 'int64".
> warnings.warn(incorrect_binary_warn % (d, name, 9000, CUDA_VERSION))
Found 33 images from 1 classes.
Class '1' (0): 33 images.
Found 125 images from 2 classes.
Class '0' (0): 44 images.
Class '1' (1): 81 images.
Found 128 images from 2 classes.
Class '0' (0): 34 images.
Class '1' (1): 94 images.
Found 128 images from 2 classes.
Class '0' (0): 34 images.
Class '1' (1): 94 images.
Found 128 images from 2 classes.
Class '0' (0): 34 images.
Class '1' (1): 94 images.
Found 128 images from 2 classes.
Class '0' (0): 34 images.
Class '1' (1): 94 images.
ERROR - resnet152-limit-125-full-basic - Failed after 0:04:32!
Traceback (most recent calls WITHOUT Sacred internals):
File "train.py", line 205, in main
save_images(datasets['samples'], to=AUGMENTED_IMAGES_DIR, n=32)
File "train.py", line 116, in save_images
save_image(dataset[i][0], img_path)
File "/home/move_eric/auglib-code/skin-data-augmentation-master/auglib/dataset_loader.py", line 48, in __getitem__
self.data.loc[index, self.image_field])
File "/home/move_eric/anaconda3/envs/torch2/lib/python3.6/posixpath.py", line 94, in join
genericpath._check_arg_types('join', a, *p)
File "/home/move_eric/anaconda3/envs/torch2/lib/python3.6/genericpath.py", line 149, in _check_arg_types
(funcname, s.__class__.__name__)) from None
TypeError: join() argument must be str or bytes, not 'int64'