missing modules

[chris201286]

Hi,

I am trying to run the fine-tunning step. It seems that there is no such 'NeuralData4Areas' module. See line 12 in fine_tuning_train.py. Is there a way to fix this?

Thank you!

[chris201286]

I could go around that by muting this line. However, I encountered some other issues: 1. line 67, data_config, data_params = load_data_config(args), 'brain_areas' should be assigned to data_params instead of data_config. 2. line 168, AttributeError: 'CANDYDynamics' object has no attribute 'dfine_list', as well as 'ldm_individual_list', 'subject_discriminator'. Should it be model._load_ckpt(model.candy_list, model.ldm, model.mapper, model.optimizer)? 3. line 152, parent_folder = f"{save_params['save_path']}/{area}/{model_name}/frac_1.0" might need to be parent_folder = f"{save_params['save_path']}/{area[0]}/{model_name}/frac_1.0". 4. after line 81, should add model_params['model_type'] = args.model_type?