-
Notifications
You must be signed in to change notification settings - Fork 27
TemporalModelCommitments #3
base: master
Are you sure you want to change the base?
Conversation
please merge upstream again, travis/awscli were giving us a hard time |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good overall, I left a few comments in the code.
Please also remove tests/__pycache__
temporal_maps[region] = time_bin_regressor | ||
return temporal_maps | ||
|
||
def _set_region_coords(self, activations, layer_regions): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I'll work on a fix for this soon, so that you hopefully won't need this here anymore (brain-score/brainio_base#1)
if type(variation) is not list: | ||
variation = [variation] | ||
assembly_dir = path.join(path.dirname(__file__), 'test_temporal_assemblies') | ||
test_assembly_filename = 'temporal_test_{}_var{}.pkl'.format(region, ''.join(str(v) for v in variation)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
you can also do f'temporal_test_{region}_var{"".join(str(v) for v in variation)}.pkl'
expected_region_count = len(expected_region) | ||
expected_time_bin_count = len(training_assembly.time_bin.values) | ||
|
||
extractor = pytorch_custom() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
use model_ctr
temporal_model = TemporalModelCommitment('', extractor, layers) | ||
# commit region: | ||
temporal_model.commit_region(region, commit_assembly) | ||
temporal_model.do_commit_region(region) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
remove
No description provided.