训练自己数据集时，显示程序已结束，need help
C:\Users\ws\anaconda3\envs\pytorch\python.exe C:\Users\ws\Desktop\lyl\edge-connect-master\train.py
Model configurations:
MODE: 1 # 1: train, 2: test, 3: eval
MODEL: 1 # 1: edge model, 2: inpaint model, 3: edge-inpaint model, 4: joint model
MASK: 3 # 1: random block, 2: half, 3: external, 4: (external, random block), 5: (external, random block, half)
EDGE: 1 # 1: canny, 2: external
NMS: 1 # 0: no non-max-suppression, 1: applies non-max-suppression on the external edges by multiplying by Canny
SEED: 10 # random seed
GPU: [0] # list of gpu ids
DEBUG: 0 # turns on debugging mode
VERBOSE: 0 # turns on verbose mode in the output console
TRAIN_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/train_flist.flist
VAL_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/val_flist.flist
TEST_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/test_flist.flist
################当EDGE=2 需要外部输入时才需要输入
TRAIN_EDGE_FLIST: ./datasets/places2_edges_train.flist
VAL_EDGE_FLIST: ./datasets/places2_edges_val.flist
TEST_EDGE_FLIST: ./datasets/places2_edges_test.flist
TRAIN_MASK_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/mask_train_flist.flist
VAL_MASK_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/mask_val_flist.flist
TEST_MASK_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/mask_test_flist.flist
LR: 0.0001 # learning rate
D2G_LR: 0.1 # discriminator/generator learning rate ratio
BETA1: 0.0 # adam optimizer beta1
BETA2: 0.9 # adam optimizer beta2
BATCH_SIZE: 8 # input batch size for training
INPUT_SIZE: 256 # input image size for training 0 for original size
SIGMA: 2 # standard deviation of the Gaussian filter used in Canny edge detector (0: random, -1: no edge)
MAX_ITERS: 2e6 # maximum number of iterations to train the model
EDGE_THRESHOLD: 0.5 # edge detection threshold
L1_LOSS_WEIGHT: 1 # l1 loss weight
FM_LOSS_WEIGHT: 10 # feature-matching loss weight
STYLE_LOSS_WEIGHT: 250 # style loss weight
CONTENT_LOSS_WEIGHT: 0.1 # perceptual loss weight
INPAINT_ADV_LOSS_WEIGHT: 0.1 # adversarial loss weight
GAN_LOSS: nsgan # nsgan | lsgan | hinge
GAN_POOL_SIZE: 0 # fake images pool size
SAVE_INTERVAL: 1000 # how many iterations to wait before saving model (0: never)
SAMPLE_INTERVAL: 1000 # how many iterations to wait before sampling (0: never)
SAMPLE_SIZE: 8 # number of images to sample
EVAL_INTERVAL: 0 # how many iterations to wait before model evaluation (0: never)
LOG_INTERVAL: 10 # how many iterations to wait before logging training status (0: never)
start training...
Training epoch: 1
进程已结束，退出代码为 -1073741819 (0xC0000005)
训练自己数据集时，显示程序已结束，need help
C:\Users\ws\anaconda3\envs\pytorch\python.exe C:\Users\ws\Desktop\lyl\edge-connect-master\train.py
Model configurations:
MODE: 1 # 1: train, 2: test, 3: eval
MODEL: 1 # 1: edge model, 2: inpaint model, 3: edge-inpaint model, 4: joint model
MASK: 3 # 1: random block, 2: half, 3: external, 4: (external, random block), 5: (external, random block, half)
EDGE: 1 # 1: canny, 2: external
NMS: 1 # 0: no non-max-suppression, 1: applies non-max-suppression on the external edges by multiplying by Canny
SEED: 10 # random seed
GPU: [0] # list of gpu ids
DEBUG: 0 # turns on debugging mode
VERBOSE: 0 # turns on verbose mode in the output console
TRAIN_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/train_flist.flist
VAL_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/val_flist.flist
TEST_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/test_flist.flist
################当EDGE=2 需要外部输入时才需要输入
TRAIN_EDGE_FLIST: ./datasets/places2_edges_train.flist
VAL_EDGE_FLIST: ./datasets/places2_edges_val.flist
TEST_EDGE_FLIST: ./datasets/places2_edges_test.flist
TRAIN_MASK_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/mask_train_flist.flist
VAL_MASK_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/mask_val_flist.flist
TEST_MASK_FLIST: C:/Users/ws/Desktop/lyl/edge-connect-master/dateset/train-dataset/flist/mask_test_flist.flist
LR: 0.0001 # learning rate
D2G_LR: 0.1 # discriminator/generator learning rate ratio
BETA1: 0.0 # adam optimizer beta1
BETA2: 0.9 # adam optimizer beta2
BATCH_SIZE: 8 # input batch size for training
INPUT_SIZE: 256 # input image size for training 0 for original size
SIGMA: 2 # standard deviation of the Gaussian filter used in Canny edge detector (0: random, -1: no edge)
MAX_ITERS: 2e6 # maximum number of iterations to train the model
EDGE_THRESHOLD: 0.5 # edge detection threshold
L1_LOSS_WEIGHT: 1 # l1 loss weight
FM_LOSS_WEIGHT: 10 # feature-matching loss weight
STYLE_LOSS_WEIGHT: 250 # style loss weight
CONTENT_LOSS_WEIGHT: 0.1 # perceptual loss weight
INPAINT_ADV_LOSS_WEIGHT: 0.1 # adversarial loss weight
GAN_LOSS: nsgan # nsgan | lsgan | hinge
GAN_POOL_SIZE: 0 # fake images pool size
SAVE_INTERVAL: 1000 # how many iterations to wait before saving model (0: never)
SAMPLE_INTERVAL: 1000 # how many iterations to wait before sampling (0: never)
SAMPLE_SIZE: 8 # number of images to sample
EVAL_INTERVAL: 0 # how many iterations to wait before model evaluation (0: never)
LOG_INTERVAL: 10 # how many iterations to wait before logging training status (0: never)
start training...
Training epoch: 1
进程已结束，退出代码为 -1073741819 (0xC0000005)