search

e-271 / few-shot-gan

Few-shot adaptation of GANs.
https://e-271.github.io/few-shot-gan
Other
120 stars 12 forks source link

Assertion Error #15

Open DebapriyaHazra opened 2 years ago

DebapriyaHazra commented 2 years ago

I am getting this error when I am trying to train the model on my dataset. I have first created the tfrecord files with dataset_tool.py.

Streaming data using training.dataset.TFRecordDataset... Traceback (most recent call last): File "run_training.py", line 254, in main() File "run_training.py", line 249, in main run(vars(args)) File "run_training.py", line 169, in run dnnlib.submit_run(kwargs) File "F:\Debapriya\Medical\Bone Marrow\Code\Thesis\few-shot-gan-master\few-shot-gan-master\dnnlib\submission\submit.py", line 343, in submit_run return farm.submit(submit_config, host_run_dir) File "F:\Debapriya\Medical\Bone Marrow\Code\Thesis\few-shot-gan-master\few-shot-gan-master\dnnlib\submission\internal\local.py", line 22, in submit return run_wrapper(submit_config) File "F:\Debapriya\Medical\Bone Marrow\Code\Thesis\few-shot-gan-master\few-shot-gan-master\dnnlib\submission\submit.py", line 280, in run_wrapper run_func_obj(submit_config.run_func_kwargs) File "F:\Debapriya\Medical\Bone Marrow\Code\Thesis\few-shot-gan-master\few-shot-gan-master\training\training_loop.py", line 151, in training_loop training_set = dataset.load_dataset(data_dir=dnnlib.convert_path(train_data_dir), verbose=True, dataset_args) File "F:\Debapriya\Medical\Bone Marrow\Code\Thesis\few-shot-gan-master\few-shot-gan-master\training\dataset.py", line 207, in load_dataset dataset = dnnlib.util.get_obj_by_name(class_name)(**kwargs) File "F:\Debapriya\Medical\Bone Marrow\Code\Thesis\few-shot-gan-master\few-shot-gan-master\training\dataset.py", line 54, in init assert os.path.isdir(self.tfrecord_dir) AssertionError