check_arch.py

import h5py

def print_model_details(file_path):
    with h5py.File(file_path, 'r') as f:
        print(f.keys())  # Print layers
        print(len(f.keys())) 
        print("")
        for key in f.keys():
            print(len(list(f[key].keys())))
            print(f"{key}: {list(f[key].keys())}")  # Print details of each layer)
print('rock')
print_model_details('model_classification/rock-170.h5')
print('mummified-170')
print_model_details('model_classification/mummified-170.h5')
print('BEiT')
print_model_details('model_classification/fossil-142.h5')
print('BEiT New')
print_model_details('model_classification/fossil-new.h5')
print("Newest:")
print_model_details('model_classification/fossil-model.h5')